chex~=0.1.86
cloudpickle~=3.0.0
flax~=0.8.5
gymnasium~=0.29.1
jax~=0.4.30
jaxlib~=0.4.30
lz4~=4.3.3
matplotlib~=3.9.1
optax~=0.2.3
tensorboardX~=2.6.2.2
wandb~=0.17.6

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

[full]
joblib~=1.4.2
pygame~=2.6.0
seaborn~=0.13.2
tensorflow~=2.17.0
