torch
pytorch_lightning
transformers
torchmetrics
jsonlines
numpy