torch
transformers[torch]