jax==0.7.0
jaxlib==0.7.0
ml_dtypes>=0.5.1
numpy>=2.3.2
opt_einsum>=3.4.0
scipy>=1.16.0
scikit-learn>=1.6.1
matplotlib>=3.10.3
