torch
transformers<6,>=5.12.1
safetensors
packaging

[ascend]

[biren]

[cuda]

[cuda:platform_system == "Linux"]
ninja

[fla]
flash-linear-attention

[fla-reference]
flash-linear-attention

[metax]

[mlx]

[mlx:platform_system == "Darwin" and platform_machine == "arm64"]
mlx

[quant]
bitsandbytes

[test]
pytest>=8.0
numpy
peft<1,>=0.19.1
accelerate
datasets
trl<2,>=1.7

[test:platform_system == "Darwin" and platform_machine == "arm64"]
mlx

[test:python_version < "3.11"]
tomli>=2.0

[torchao]

[torchao:platform_system == "Linux"]
torchao

[train]
peft<1,>=0.19.1
accelerate
datasets
trl<2,>=1.7
deepspeed
