numpy>=1.20.0
scipy>=1.7.0
requests>=2.25.0
tqdm>=4.60.0
pyyaml>=5.4.0
pandas>=1.3.0
matplotlib>=3.3.0
click>=8.0.0
joblib>=1.0.0
h5py>=3.0.0

[all]
cupy>=10.0.0
jax>=0.3.0
jaxlib>=0.3.0
torch>=1.9.0

[cupy]
cupy>=10.0.0

[dev]
pytest>=7.0.0
pytest-cov>=3.0.0
black>=22.0.0
flake8>=4.0.0
mypy>=0.950
sphinx>=4.5.0

[jax]
jax>=0.3.0
jaxlib>=0.3.0

[torch]
torch>=1.9.0
