numpy
scipy
jax
jaxlib
matplotlib
