torch

[:platform_system == "Linux"]
triton

[all]
cut-cross-entropy[transformers]
accelerate>=0.34.2
datasets>=3.1.0
huggingface_hub>=0.26.2
pandas
fire
tqdm

[all:platform_system != "Darwin"]
deepspeed>=0.15.1

[dev]
cut-cross-entropy[all,test]
build
twine
pre-commit

[test]
pytest

[transformers]
transformers>=4.44.2
