## Setup:
1. you need have cuda installed
2. run the auto_setup.sh script to setup the conda environment
3. `conda activate jax`
4. longin to wandb

## Usage
The main entry point is the `./scripts/shell/brax_experiment.py` python script.
The type of experiment can be set why argparser variables for `-env -algo -seed -gpu -H`
An example call would be `python brax_experiment.py -env inverted_pendulum -algo SAC-MVE -H 0 -seed 1 -wandb_mode disabled`

