numpy>=1.20.0

[all]
numpy
matplotlib
plotly

[cuda]

[cuda11]
cupy-cuda11x>=12.0.0

[cuda118]
cupy-cuda118>=12.0.0

[cuda12]
cupy-cuda12x>=12.0.0

[cuda120]
cupy-cuda120>=12.0.0

[dev]
pytest>=7.0.0
ruff>=0.1.0
mypy>=1.0.0
build>=1.0.0
twine>=4.0.0

[docs]
sphinx>=7.0.0
sphinx-rtd-theme>=2.0.0
myst-parser>=2.0.0

[rocm]

[torch]
torch>=1.9

[visualization]
matplotlib
plotly
