jax>=0.4.11
flax>=0.7.2
chex>=0.1.7
dotmap>=1.3.30
evosax>=0.1.4
matplotlib>=3.3.4
numpy>=1.22.4
distrax>=0.1.3
pillow>=9.5.0
brax>=0.9
gymnax>=0.0.6
typing-extensions<4.6.0
wandb
hydra-core>=1.3.2
omegaconf>=2.3.0
safetensors>=0.3.3
optax>=0.1.4
imageio>=2.33.1
