numpy
typing-extensions
jax
jaxlib
matplotlib
scipy
astropy
