jax>=0.4.30
flax>=0.10.0
optax>=0.2.0
orbax-checkpoint>=0.11.0
mctx>=0.0.6
flashbax>=0.1.3
gymnasium>=1.1.0
numpy>=2.0.0
tqdm
rlax>=0.1.9
distrax>=0.1.9
tyro>=1.0.16

[cuda12]
jax[cuda12]>=0.4.30

[wandb]
wandb
