torch
tokenizers
lightning
torchmetrics
huggingface_hub