equinox!=0.13.3,>=0.13.1
interpax_fft>=0.0.8
jax!=0.5.1,!=0.5.2,!=0.6.0,!=0.7.1,<=0.8.2,>=0.5.0
lineax
matplotlib
numpy
optax
optimistix
orbax-checkpoint
orthax
paramax
