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

[onnx]
onnxruntime>=1.18
tokenizers>=0.20
jinja2>=3.1

[onnx-export]
jul[torch]
onnx>=1.16
onnxscript

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

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

[tune]
scikit-learn>=1.3
scipy>=1.10

[yaml]
pyyaml>=6
