numpy
transformers>=5.0.0
sentencepiece
protobuf
pyyaml
jinja2

[:platform_system == "Darwin"]
mlx>=0.30.4

[cpu]
mlx[cpu]>=0.30.4

[cuda12]
mlx[cuda12]>=0.30.4

[cuda13]
mlx[cuda13]>=0.30.4

[evaluate]
lm-eval
tqdm

[test]
datasets
lm-eval

[train]
datasets
tqdm
