numpy>=2.0.0
einops>=0.6.0
autoray>=0.7.0
matplotlib>=3.8.0
jax>=0.4.34
equinox>=0.11.4
lrux>=0.1.0
