transformers>=5.5.0
accelerate
datasets
lm-eval
primefac
scipy
gemlite>=0.5.1
hqq>=0.2.8
safetensors
ortools>=9.15.6755

[cpu]

[cpu:sys_platform == "linux"]
torch
torchvision

[cu118]

[cu118:sys_platform == "linux"]
torch
torchvision

[cu121]

[cu121:sys_platform == "linux"]
torch
torchvision

[cu124]

[cu124:sys_platform == "linux"]
torch
torchvision

[cu126]

[cu126:sys_platform == "linux"]
torch
torchvision

[cu128]

[cu128:sys_platform == "linux"]
torch
torchvision

[cu130]

[cu130:sys_platform == "linux"]
torch
torchvision

[dev]
black
hydra-core
pre-commit
pylint
pytest

[distributed]
deepspeed

[docs]
mkdocs-material
mkdocstrings[python]
mkdocs-gen-files
mkdocs-literate-nav

[eval]
hydra-core>=1.3
omegaconf>=2.3
openai>=1.0
matplotlib>=3.8
japanize-matplotlib>=1.1
shortuuid>=1.0

[hydra]
hydra-core

[mps]

[mps:sys_platform == "darwin"]
torch
torchvision

[visualize]
matplotlib>=3.10.8

[vllm]
vllm<0.22,>=0.10
