torch>=2.0.0
tqdm>=4.65.0

[dev]
pytest>=7.0.0
build>=1.4.0
twine>=6.2.0

[docs]
mkdocs-material>=9.0.0

[examples]
transformers>=4.30.0
datasets>=2.0.0
accelerate>=0.20.0
numpy>=1.20.0
python-dotenv>=1.0.0
pyyaml>=6.0
matplotlib>=3.9.0

[mlx]
mlx>=0.20.0
mlx-lm>=0.19.0
