ml_dtypes
numpy
pipescript>=0.0.25
pyccolo>=0.0.90

[backends]
black<24
einops
ipython
isort
mypy
pytest
pytest-cov
ruff
torch
jax
tensorflow
onnx
onnxscript
onnxruntime

[cupy]
cupy

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

[einops]
einops

[jax]
jax

[onnx]
torch
onnx
onnxscript
onnxruntime

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

[tf]
tensorflow

[torch]
torch
