torch >= 2.0
typing_extensions
