einops<1,>=0.7.0
safetensors<1,>=0.4.0
torch<3.0,>=2.0

[mlx]
mlx<1,>=0.10.0

[test]
black==24.4.2
mlx==0.10.0
mypy==1.10.0
numpy==1.26.4
pre-commit==3.7.0
pytest==8.1.2
pytest-cov==5.0.0
ruff==0.4.2
sentencepiece==0.2.0

[train]
lora-pytorch~=0.1.0
lightning~=2.2.0
tensorboard~=2.14.0
torchdata~=0.7.0
tqdm
