torch
torchmetrics
transformers
