setuptools
pandas
numpy
scipy
six
wheel
jax[cuda]==0.4.11
jaxlib==0.4.11
distrax
brax
chex
flax
optax
gym
notebook
matplotlib
tqdm
gymnax
jupyter
ipython
wandb
