jax>=0.4.13
jaxtyping>=0.2.20
typing_extensions>=4.5.0
equinox>=0.11.3
matplotlib>=3.8.1
