torch
tensorflow-gpu
transformers