torch>=1.9.0
typing_extensions
scikit-learn