aim
brax
chex
distrax
flax
tfp-nightly[jax]
gymnasium>=1.1.0
hydra-core
hydra-joblib-launcher
jax>=0.4.35
jaxlib>=0.4.35
numpy
omegaconf
optax
orbax-checkpoint
pandas
scipy
typing-extensions

[comet]
comet_ml

[dev]
furo
myst-parser
pre-commit
pytest
ruff
sphinx
sphinx-autodoc2
sphinx-copybutton

[envpool]
envpool

[gymnasium]
gymnasium[atari,box2d,classic-control,mujoco]>=1.1.0

[gymnax]
gymnax

[jaxmarl]
jaxmarl

[jumanji]
jumanji

[mujoco-playground]
playground

[neptune]
neptune-scale

[swanlab]
swanlab

[wandb]
wandb
