jax>=0.4.13
jaxlib>=0.4.13
flax>=0.7.0
rich>=13.4.2

[baselines]
matplotlib>=3.7.2
imageio>=2.31.2
imageio-ffmpeg>=0.4.9
wandb>=0.15.10
pyrallis>=0.3.1
distrax>=0.1.4
optax>=0.1.5
orbax>=0.1.9

[dev]
ruff>=0.1.6
pre-commit>=3.3.3
