jax>=0.6.0
jaxlib>=0.6.0
numpy>=2.1.0
optax>=0.2.4
tqdm>=4.67.1
scikit-learn>=1.6.1

[gpu]
jax[cuda12]>=0.6.0

[tpu]
jax[tpu]>=0.6.0
