BraxEnv

torchrl.envs.libs.brax.BraxEnv(*args, _inplace_update=False, _batch_locked=True, **kwargs)

Google Brax environment wrapper.

Examples

>>> env = BraxEnv(env_name="ant")
>>> td = env.rand_step()
>>> print(td)
>>> print(env.available_envs)