pytorch_lightning
torch
transformers