numpy
scikit-learn>=1.2.2
jax[cpu]
scipy
