jax>=0.8
jaxlib>=0.8
diffrax
matplotlib
numpy
scipy
