mjx-safety-gym 0.1.0

MJX Safety Gym Environments

mjx-safety-gym

Setup

Create a Python venv with version 3.11 (or above).

Then, pip install -r requirements.txt After whcih pip install -e . To install the module

If you want to use vision-based observations, see the Madrona section first.

How to Run

Open-source MJX implementation of OpenAI Safety Gym for accelerated safe reinforcement learning.

To run the interactive viewer on MacOS M1, run sudo mjpython scripts/interactive.py

Note that we need sudo privileges to get access to keypress information.

Madrona

To use vision-based observations, run

chmod +x vision_setup.bash
./vision_setup.bash

This requires a linux machine with an NVidia GPU and can take up to 5 minutes to install.