torch>=2.0
numpy>=1.21

[dev]
pytest>=7.0
