packaging>=23.0
rich>=13.0
psutil>=5.9
scikit-learn>=1.3
huggingface-hub>=0.20

[all]
neuralops[benchmark,cuda,docs,eval,export,mlops]

[benchmark]
matplotlib>=3.7
pandas>=2.0
seaborn>=0.13
rich>=13.0.0
psutil>=5.9.0

[core]
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
rich>=13.0
psutil>=5.9
scikit-learn>=1.3
huggingface-hub>=0.20

[cpu]
neuralops[core]

[cuda]
neuralops[core]

[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

[evaluation-experiments]
transformers>=4.40.0
numpy>=1.24.0

[export]
safetensors>=0.4
onnx>=1.15
onnxruntime>=1.17

[fla]

[fla:sys_platform == "linux"]
fla-linear-attention<0.3,>=0.2.0

[flash_attn]

[flash_attn:sys_platform == "linux"]
flash_attn

[mamba]

[mamba:sys_platform == "linux"]
mamba_ssm

[mlops]
wandb>=0.16
mlflow>=2.5
tensorboard>=2.15
accelerate>=0.25

[mps]
neuralops[core]

[optuna]
optuna

[rocm]
neuralops[core]

[serving]
transformers>=4.40.0
fastapi>=0.110.0
uvicorn[standard]>=0.27.0

[tensorrt_llm]
tensorrt_llm

[test]
neuralops[dev]

[trl]
trl

[unsloth]

[unsloth:sys_platform == "linux"]
unsloth

[vllm]
vllm
