# Core
equinox
jax
jaxlib
jaxtyping
numpy # Needed for serialisation
optax # Needed for optimisation