jax>=0.4.20
numpy>=1.24

[diffrax]
diffrax>=0.5

[examples]
matplotlib>=3.6

[test]
pytest>=7
