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