torch>=2.6.0
numpy
safetensors>=0.4.3
typing_extensions>=4.6.0

[dev]
pytest>=7.0.0
pytest-cov>=4.0.0
expecttest
black>=23.0.0
isort>=5.12.0
mypy>=1.5.0

[test]
pytest>=7.0.0
pytest-cov>=4.0.0
expecttest
