chex~=0.1.88
cloudpickle~=3.1.0
flax~=0.10.2
gymnasium~=1.0.0
jax~=0.4.38
lz4~=4.3.3
matplotlib~=3.9.4
optax~=0.2.4
tensorboardX~=2.6.2.2
wandb~=0.19.2

[docs]
sphinx~=8.1.3
sphinx-rtd-theme~=3.0.2

[full]
joblib~=1.4.2
pygame~=2.6.1
seaborn~=0.13.2
tensorflow~=2.18.0
