numpy>=1.15
matplotlib>=3.4
jax>=0.2.10