matplotlib
numpy
torch>=2.9
scipy
tqdm

[scimba_jax]
jax>=0.6.2
equinox>=0.13.0
optax
jax-tqdm
