torch>=2.2
numpy>=1.20
typing_extensions>=4.0

[all]
pytest>=7.0
pytest-cov>=4.0
tqdm>=4.60.0
numpy<2
optree
jax[cuda12]==0.4.27
neural-tangents==0.6.5

[benchmark]
tqdm>=4.60.0

[dev]
pytest>=7.0
pytest-cov>=4.0
tqdm>=4.60.0

[parity]
pytest>=7.0
pytest-cov>=4.0
numpy<2
optree
jax[cuda12]==0.4.27
neural-tangents==0.6.5

[test]
pytest>=7.0
pytest-cov>=4.0
