torch>=2.1.0
transformers>=4.40.0
accelerate>=0.28.0

[cuda]
bitsandbytes>=0.43.0

[dev]
pytest>=7.0
datasets>=2.18.0
matplotlib>=3.8.0
tqdm>=4.66.0

[mlx]
mlx>=0.12.0
mlx-lm>=0.12.0
