numpy
scipy
scikit-learn
matplotlib
jax
jaxlib