torchmetrics
TorchMetrics provides over 100 PyTorch-compatible metrics with features like automated synchronization, accumulation, and built-in visualization tools. Designed for distributed training, it integrates seamlessly with PyTorch Lightning to minimize coding overhead. Ideal for machine learning tasks in domains like classification and regression across multi-device setups.
