Audience
Anyone seeking a solution providing several PyTorch metrics implementations to create custom metrics
About TorchMetrics
TorchMetrics is a collection of 90+ PyTorch metrics implementations and an easy-to-use API to create custom metrics. A standardized interface to increase reproducibility. It reduces boilerplate. distributed-training compatible. It has been rigorously tested. Automatic accumulation over batches. Automatic synchronization between multiple devices. You can use TorchMetrics in any PyTorch model, or within PyTorch Lightning to enjoy additional benefits. Your data will always be placed on the same device as your metrics. You can log Metric objects directly in Lightning to reduce even more boilerplate. Similar to torch.nn, most metrics have both a class-based and a functional version. The functional versions implement the basic operations required for computing each metric. They are simple python functions that as input take torch.tensors and return the corresponding metric as a torch.tensor. Nearly all functional metrics have a corresponding class-based metric.
Other Popular Alternatives & Related Software
PyTorch
Transition seamlessly between eager and graph modes with TorchScript, and accelerate the path to production with TorchServe. Scalable distributed training and performance optimization in research and production is enabled by the torch-distributed backend. A rich ecosystem of tools and libraries extends PyTorch and supports development in computer vision, NLP and more. PyTorch is well supported on major cloud platforms, providing frictionless development and easy scaling. Select your preferences and run the install command. Stable represents the most currently tested and supported version of PyTorch. This should be suitable for many users. Preview is available if you want the latest, not fully tested and supported, 1.10 builds that are generated nightly. Please ensure that you have met the prerequisites (e.g., numpy), depending on your package manager. Anaconda is our recommended package manager since it installs all dependencies.
Learn more
AWS Neuron
It supports high-performance training on AWS Trainium-based Amazon Elastic Compute Cloud (Amazon EC2) Trn1 instances. For model deployment, it supports high-performance and low-latency inference on AWS Inferentia-based Amazon EC2 Inf1 instances and AWS Inferentia2-based Amazon EC2 Inf2 instances. With Neuron, you can use popular frameworks, such as TensorFlow and PyTorch, and optimally train and deploy machine learning (ML) models on Amazon EC2 Trn1, Inf1, and Inf2 instances with minimal code changes and without tie-in to vendor-specific solutions. AWS Neuron SDK, which supports Inferentia and Trainium accelerators, is natively integrated with PyTorch and TensorFlow. This integration ensures that you can continue using your existing workflows in these popular frameworks and get started with only a few lines of code changes. For distributed model training, the Neuron SDK supports libraries, such as Megatron-LM and PyTorch Fully Sharded Data Parallel (FSDP).
Learn more
ExecuTorch
ExecuTorch is PyTorch’s open source framework for deploying AI/ML models directly to edge devices, enabling text, vision, speech, recommendation, and multimodal inference without requiring the cloud. It exports models from PyTorch without intermediate conversion formats, preserves ATen operators, and uses ahead-of-time compilation to optimize performance for target hardware before deployment. Its modular design lets developers choose compile-time and runtime optimizations while staying inside the familiar PyTorch ecosystem, including torchao for quantization. A portable C++ runtime with a base footprint of about 50 KB can run on smartphones, desktops, embedded systems, microcontrollers, DSPs, and Cortex-M processors. ExecuTorch supports Android, iOS, Linux, Windows, macOS, and WebAssembly, with native APIs for C++, Swift, Kotlin, and Objective-C.
Learn more
Google Cloud Deep Learning VM Image
Provision a VM quickly with everything you need to get your deep learning project started on Google Cloud. Deep Learning VM Image makes it easy and fast to instantiate a VM image containing the most popular AI frameworks on a Google Compute Engine instance without worrying about software compatibility. You can launch Compute Engine instances pre-installed with TensorFlow, PyTorch, scikit-learn, and more. You can also easily add Cloud GPU and Cloud TPU support. Deep Learning VM Image supports the most popular and latest machine learning frameworks, like TensorFlow and PyTorch. To accelerate your model training and deployment, Deep Learning VM Images are optimized with the latest NVIDIA® CUDA-X AI libraries and drivers and the Intel® Math Kernel Library. Get started immediately with all the required frameworks, libraries, and drivers pre-installed and tested for compatibility. Deep Learning VM Image delivers a seamless notebook experience with integrated support for JupyterLab.
Learn more
Pricing
Starting Price:
Free
Free Version:
Free Version available.
Company Information
TorchMetrics
United States
torchmetrics.readthedocs.io/en/stable/
Videos and Screen Captures
Other Useful Business Software
Ship Agents Faster
Gemini Enterprise Agent Platform lets you rapidly build, scale, govern and optimize production-ready agents grounded in your organization's data. The platform enables developers to build custom or pre-built agents for virtually any use case. New customers get $300 in free credits.
Product Details
Platforms Supported
Cloud
Training
Documentation
Support
Online
