jax
jaxlib
matplotlib>=3.1.2
numpy>=1.17.4
