numpy<3.0,>=1.24.0
scipy<2.0,>=1.10.0
scikit-learn<2.0,>=1.3.0
jax<1.0,>=0.4.25
jaxlib<1.0,>=0.4.25
optax<1.0,>=0.1.7
chex<1.0,>=0.1.8
typing-extensions<5.0,>=4.5.0
