TorchMetrics — a short history to 100+ metrics
Blog post from Lightning AI
TorchMetrics evolved from an early PyTorch Lightning metric-interface proposal in April 2020, created to provide unified, distributed-aware metric computation and initially relying heavily on scikit-learn for correctness and rapid coverage. Early development introduced separate NumPy and tensor metric handling, distributed aggregation, native PyTorch implementations for common classification metrics, and contributions such as MetricCollection and substantial test-speed improvements. A major API redesign in PyTorch Lightning 0.10.0 established the durable update-compute model, allowing metrics to maintain memory-efficient internal states, process batched datasets, and synchronize correctly across distributed GPUs. The metrics code became the standalone torchmetrics package in February 2021 while retaining its commit history, and it expanded from roughly 20 regression and classification metrics to more than 100 covering areas including audio, detection, image, retrieval, and text. Its contributors published a JOSS paper in February 2022, and version 1.0.0, released July 5, 2023, added plotting and visualization capabilities while marking a milestone in the project’s open-source development.
| Trend | Post Mentions | Total Month Mentions | Posts | Companies | MoM |
|---|---|---|---|---|---|
| Vector Search | 1 | 1,164 | 176 | 76 | -27% |
Use this post, company, and trend context to find content marketing opportunities, perform competitive analysis, or address product feature gaps via the Plushcap MCP server or the Plushcap API.