Torchmetrics - Machine learning metrics for distributed, scalable PyTorch applications. - View it on GitHub