torch>=2.4.0

[dev]
pytest
black==24.2.0
