GitHub repository: martin-kokos/torchmetrics
Description: Machine learning metrics for distributed, scalable PyTorch applications.