numpy
transformers==5.0.0rc1
sentencepiece
protobuf
pyyaml
jinja2

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

[cpu]
mlx[cpu]>=0.29.2

[cuda]
mlx[cuda]>=0.29.2

[evaluate]
lm-eval
tqdm

[test]
datasets
lm-eval

[train]
datasets
tqdm
