torch>=2.2
safetensors>=0.4
numpy

[test]
pytest>=8
