jax>=0.4.27
jaxlib>=0.4.27
flax>=0.8.0
rich>=13.4.2
tqdm>=4.67.1
chex>=0.1.85
imageio>=2.31.2
imageio-ffmpeg>=0.4.9

[baselines]
matplotlib>=3.7.2
wandb>=0.15.10
pyrallis>=0.3.1
distrax>=0.1.4
optax>=0.1.5
orbax>=0.1.9
tqdm>=4.66.4

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