jax>=0.4.16
jaxlib>=0.3.0
optax>=0.1.0
pyarrow>=10.0.0
orbax-checkpoint>=0.5.0

[distributed]
ray>=2.0.0

[envpool]
envpool>=0.8.0

[full]
gymnasium>=0.29.0
ray>=2.0.0
envpool>=0.8.0
gpjax>=0.8.0
plotly>=5.0.0
pandas>=2.0.0
tensorflow-datasets>=4.0.0
grain>=0.1.0
brax>=0.11.0

[gp]
gpjax>=0.8.0

[gymnasium]
gymnasium>=0.29.0

[neuroevolution]
tensorflow-datasets>=4.0.0
grain>=0.1.0
brax>=0.11.0

[test]
chex>=0.1.0
flax>=0.5.0
pytest>=6.0.0
tensorflow>=2.12.0

[vis]
plotly>=5.0.0
pandas>=2.0.0
