Environment Setup

[Ozzey]

Hello! Amazing work with the jax implementation and vectorization. I wanted to ask could you please give me some pointers on how to use this implementation for custom gym/ mujoco-gym env? Assuming it's continuous control with a simple vector for the state.

[ShaneFlandermeyer]

Using custom gym envs is pretty straightforward with the current setup! All you need to do is register the env (see here), then change the env_id field in the config file to whatever ID you use in gym.register().

[Ozzey]

Using custom gym envs is pretty straightforward with the current setup! All you need to do is register the env (see here), then change the env_id field in the config file to whatever ID you use in gym.register().

Great! Thank you for the response. Closing the issue.