jax>=0.4.28
jaxlib>=0.4.28
flax>=0.8.5
qdax>=0.4.0
tqdm>=4.66.5
imageio>=2.35.1

[cuda12]
jax[cuda12]>=0.4.30
