chex~=0.1.86
cloudpickle~=3.0.0
flax~=0.8.2
gymnasium~=0.29.1
jax~=0.4.26
jaxlib~=0.4.26
lz4~=4.3.3
matplotlib~=3.8.4
optax~=0.2.2
tensorboardX~=2.6.2.2
wandb~=0.16.6

[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
