numpy>=1.23

[coverage]
coverage[toml]>=7

[jax]
jax>=0.4.35

[sklearn]
scikit-learn>=1.6

[test]
pytest>=9
pytest-xdist>=3.6
scipy
mpmath

[test-jax]
pytest>=9
pytest-xdist>=3.6
scipy
mpmath
jax>=0.4.35

[test-torch]
pytest>=9
pytest-xdist>=3.6
scipy
mpmath
torch>=2.1
ninja
scikit-learn>=1.6

[torch]
torch>=2.1
ninja
