chex~=0.1.85
cloudpickle~=3.0.0
dm-haiku~=0.0.11
gymnasium~=0.29.1
jax~=0.4.21
jaxlib~=0.4.21
lz4~=4.3.2
matplotlib~=3.8.2
optax~=0.1.7
tensorboardX~=2.6.2.2

[docs]
sphinx~=7.2.6
sphinx-rtd-theme~=2.0.0

[full]
joblib~=1.3.2
pygame~=2.5.2
seaborn~=0.13.0
tensorflow~=2.15.0
