jax<=0.4.38
jaxlib
flax
safetensors
chex
brax==0.10.3
mujoco==3.1.3
matplotlib
pillow
scipy<=1.12
gymnax

[algs]
optax
distrax
flashbax==0.1.0
wandb
hydra-core>=1.3.2
omegaconf>=2.3.0
pettingzoo>=1.24.3
tqdm>=4.66.0

[dev]
pytest
pygame
