requirements.txt

absl-py==2.1.0
black==23.12.1
chex==0.1.85
click==8.1.7
cloudpickle==3.0.0
contourpy==1.2.0
cycler==0.12.1
decorator==5.1.1
distrax==0.1.5
dm-env==1.6
dm-tree==0.1.8
etils==1.5.2
Farama-Notifications==0.0.4
filelock==3.13.1
flax==0.7.5
fonttools==4.47.2
fsspec==2023.12.2
gast==0.5.4
gym==0.26.2
gym-notices==0.0.8
gymnasium==0.29.1
gymnax==0.0.6
importlib-metadata==7.0.1
importlib-resources==6.1.1
jax==0.4.23
jaxlib==0.4.23
Jinja2==3.1.3
joblib==1.3.2
kiwisolver==1.4.5
markdown-it-py==3.0.0
MarkupSafe==2.1.4
matplotlib==3.8.2
mdurl==0.1.2
ml-dtypes==0.3.2
mpmath==1.3.0
msgpack==1.0.7
mypy-extensions==1.0.0
nest-asyncio==1.6.0
networkx==3.2.1
numpy==1.26.3
opt-einsum==3.3.0
optax==0.1.8
orbax-checkpoint==0.5.0
packaging==23.2
pandas==2.2.0
pathspec==0.12.1
pillow==10.2.0
platformdirs==4.1.0
protobuf==4.25.2
pydot==2.0.0
pygments==2.17.2
pyparsing==3.1.1
python-dateutil==2.8.2
pytz==2023.3.post1
PyYAML==6.0.1
rich==13.7.0
rlax==0.1.6
scikit-learn==1.2.2
scipy==1.12.0
seaborn==0.13.1
six==1.16.0
stable-baselines3==2.2.1
sympy==1.12
tensorflow-probability==0.23.0
tensorstore==0.1.52
threadpoolctl==3.2.0
tomli==2.0.1
toolz==0.12.0
torch==2.1.2
tqdm==4.66.1
typing-extensions==4.9.0
tzdata==2023.4
zipp==3.17.0