numpy
scipy
matplotlib
jax
jaxlib
