torch>=1.11.0
typing_extensions>=4.1.0
