jax[cuda13]>=0.10.2
equinox
jaxtyping
numpy
spineax>=0.0.3
jax2sympy>=0.0.2
