numpy>=1.24

[calibrate]
datasets
pyarrow

[dev]
pytest>=8
pyyaml>=6

[lab]
jul[mlx]
pyyaml>=6
datasets
pyarrow

[mlx]

[mlx:sys_platform == "darwin" and platform_machine == "arm64"]
mlx>=0.32
mlx-lm>=0.31

[repro]
jul[mlx]
pyyaml>=6
datasets
pyarrow
pytest>=8
scikit-learn>=1.3

[torch]
torch>=2.4
transformers>=5.0
accelerate

[yaml]
pyyaml>=6
