numpy
scipy
jax
jaxlib
grgrlib>=0.1.9
pyyaml
