ml_dtypes
numpy
pipescript>=0.0.24
pyccolo>=0.0.89

[cupy]
cupy

[dev]
build
pycln
twine
setuptools-git-versioning
versioneer
black<24
ipython
isort
mypy
pytest
pytest-cov
ruff

[jax]
jax

[onnx]
torch
onnx
onnxruntime

[test]
black<24
ipython
isort
mypy
pytest
pytest-cov
ruff

[tf]
tensorflow

[torch]
torch
