GitHub - Lightning-AI/metrics: Machine learning metrics for distributed, scalable PyTorch applications.

Machine learning metrics for distributed, scalable PyTorch applications. TorchMetrics is a collection of 80 PyTorch metrics implementations and an easy-to-use API to create custom metrics. The module-based metrics contain internal metric states (similar to the parameters of the PyTorch module) that automate accumulation and synchronization across devices! This can be run on CPU, single GPU or multi-GPUs! Module metric usage remains the same when using multiple GPUs or multiple nodes.