torch>=2.0.0

[dev]
pytest>=7.0.0
yapf>=0.40.0
