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