packaging>=23.0

[all]
neuralops[export,mlops]

[benchmark]
matplotlib>=3.7.0
seaborn>=0.12.0
pandas>=2.0.0

[cpu]
torch>=2.2.0
transformers>=4.40.0
datasets>=2.18.0
einops>=0.7.0
tqdm>=4.66.0
pyyaml>=6.0
safetensors>=0.4.0
pandas>=2.0.0
scipy>=1.11.0
numpy>=1.24.0
packaging>=23.0

[cuda]
torch>=2.2.0
transformers>=4.40.0
datasets>=2.18.0
einops>=0.7.0
tqdm>=4.66.0
pyyaml>=6.0
safetensors>=0.4.0
pandas>=2.0.0
scipy>=1.11.0
numpy>=1.24.0
packaging>=23.0

[cuda:sys_platform == "linux"]
triton>=2.3.0

[dev]
pytest>=7.4.0
pytest-cov>=4.1.0
pytest-xdist>=3.5.0
ruff>=0.4.0
black>=24.0.0
isort>=5.13.0
mypy>=1.10.0
pre-commit>=3.7.0
build>=1.2.0
twine>=5.0.0
pdoc3>=0.10.0

[docs]
mkdocs>=1.5.0
mkdocs-material>=9.5.0
mkdocstrings[python]>=0.24.0
pdoc3>=0.10.0

[eval]
scikit-learn>=1.3.0
evaluate>=0.4.0
lm-eval>=0.4.0

[export]
safetensors
onnx
onnxruntime

[mlops]
wandb
mlflow
tensorboard
accelerate

[mps]
torch>=2.2.0
transformers>=4.40.0
datasets>=2.18.0
einops>=0.7.0
tqdm>=4.66.0
pyyaml>=6.0
safetensors>=0.4.0
pandas>=2.0.0
scipy>=1.11.0
numpy>=1.24.0
packaging>=23.0

[rocm]
torch>=2.2.0
transformers>=4.40.0
datasets>=2.18.0
einops>=0.7.0
tqdm>=4.66.0
pyyaml>=6.0
safetensors>=0.4.0
pandas>=2.0.0
scipy>=1.11.0
numpy>=1.24.0
packaging>=23.0
