chex~=0.1.7
cloudpickle~=2.2.1
dm-haiku~=0.0.10
gymnasium~=0.29.0
jax~=0.4.13
jaxlib~=0.4.13
lz4~=4.3.2
matplotlib~=3.7.2
optax~=0.1.5
tensorboardX~=2.6.1

[docs]
sphinx~=7.0.1
sphinx-rtd-theme~=1.2.2

[full]
joblib~=1.3.1
pygame~=2.5.0
seaborn~=0.12.2
tensorflow~=2.13.0
