torchmetrics

Subscription: Developer

Last updated an hour ago

Active

Subscription: Developer

A PyTorch-native library providing a collection of machine learning metrics for classification, regression, detection, and NLP tasks, with support for distributed synchronization, metric aggregation, and integration with Lightning trainers.