numpy
transformers==5.0.0rc3
sentencepiece
protobuf
pyyaml
jinja2

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

[cpu]
mlx[cpu]>=0.30.3

[cuda12]
mlx[cuda12]>=0.30.3

[cuda13]
mlx[cuda13]>=0.30.3

[evaluate]
lm-eval
tqdm

[test]
datasets
lm-eval

[train]
datasets
tqdm
