jax>=0.4.20
jaxlib>=0.4.20
equinox>=0.11.0
jaxtyping>=0.2.20
plum-dispatch>=2.0.0
diffrax>=0.5.0
einops>=0.7.0
matplotlib>=3.8.0

[test]
pytest>=8.3.5
