torch>=1.9.0

[all]
pytest>=7.0
ruff>=0.4
mypy>=1.0
torchvision>=0.10
transformers>=4.30
lightning>=2.0
datasets>=2.0
numpy>=1.20

[all:platform_system != "Darwin"]
triton>=2.1

[benchmark]
numpy>=1.20
torchvision>=0.10
transformers>=4.30
datasets>=2.0

[cuda]

[cuda:platform_system != "Darwin"]
triton>=2.1

[dev]
pytest>=7.0
ruff>=0.4
mypy>=1.0
torchvision>=0.10

[hf]
transformers>=4.30
accelerate>=0.20
safetensors>=0.3
tokenizers>=0.13

[integrations]
transformers>=4.30
accelerate>=0.20
safetensors>=0.3
tokenizers>=0.13
lightning>=2.0
