numpy>=1.26.4
scikit-learn>=1.3.2
tqdm>=4.67.1

[all]
jax[cuda13]>=0.10.0
torch>=2.5.0
cupy-cuda12x>=13.4.1

[cupy]
cupy-cuda12x>=13.4.1

[dev]
pytest>=7.0
fbm>=0.3.0

[jax-cpu]
jax[cpu]>=0.4.34

[jax-gpu]
jax[cuda13]>=0.10.0

[torch]
torch>=2.5.0
