jax<0.11,>=0.10.2
equinox<0.14,>=0.13.8
jaxtyping>=0.2
numpy>=2.0
safetensors>=0.4

[convert]
torch>=2.6

[cuda13]
jax[cuda13]<0.11,>=0.10.2

[dev]
pytest>=7.0
torch>=2.6
ruff>=0.5
