torch>=2.1.0
pytest>=7.0.0
