torch
transformers[torch]
scikit-learn
datasets

[dev]
check-manifest
pytest

[test]
coverage
