jax>=0.1.55
jaxlib>=0.1.37
numpy>=1.18.0
scipy
matplotlib

[dev]

[test]
pytest
pytest-cov
