torch
pytorch_lightning
transformers
numpy