torch
transformers
pytorch-lightning
