torch>=1.9.0
numpy>=1.18.0
pytest>=6.0.0
