jax
jaxlib
numpy
matplotlib
astropy
