gymnasium>=0.26
numpy<2.0.0
networkx>=2.4
matplotlib>=3.1
pyyaml>=5.3
prettytable>=0.7
safetensors>=0.4.2
orbax-checkpoint==0.5.0
flax==0.10.7
gymnax==0.0.9
wandb==0.24.1
distrax==0.1.5
optax==0.2.5
chex==0.1.90
flashbax==0.1.3
plotext
hydra-core
jax[cuda12]==0.6.2

[docs]
mkdocs==1.6.1
mkdocstrings==1.0.3
mkdocstrings[python]
