numpy
matplotlib
jax
