chex~=0.1.85
cloudpickle~=3.0.0
flax~=0.8.1
gymnasium~=0.29.1
jax~=0.4.24
jaxlib~=0.4.24
lz4~=4.3.3
matplotlib~=3.8.2
optax~=0.1.9
tensorboardX~=2.6.2.2
wandb~=0.16.3

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

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