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