chex>=0.1.91
flax>=0.12.0
jax>=0.8.0
plotly>=6.4.0

[gpu]
jax[cuda12]>=0.8.0
