Average Ratings 0 Ratings
Average Ratings 0 Ratings
Description
TorchMetrics comprises over 90 implementations of metrics designed for PyTorch, along with a user-friendly API that allows for the creation of custom metrics. It provides a consistent interface that enhances reproducibility while minimizing redundant code. The library is suitable for distributed training and has undergone thorough testing to ensure reliability. It features automatic batch accumulation and seamless synchronization across multiple devices. You can integrate TorchMetrics into any PyTorch model or utilize it within PyTorch Lightning for added advantages, ensuring that your data aligns with the same device as your metrics at all times. Additionally, you can directly log Metric objects in Lightning, further reducing boilerplate code. Much like torch.nn, the majority of metrics are available in both class-based and functional formats. The functional versions consist of straightforward Python functions that accept torch.tensors as inputs and yield the corresponding metric as a torch.tensor output. Virtually all functional metrics come with an equivalent class-based metric, providing users with flexible options for implementation. This versatility allows developers to choose the approach that best fits their coding style and project requirements.
Description
tensorViz provides a platform for developers to delve into PyTorch models and JAX/Flax programs through dynamic graph visualizations. In the case of PyTorch, users have the ability to open nested modules, examine tensor shapes recorded during a single forward pass, navigate to the corresponding Python source code, and observe architectural modifications alongside code differences. The site additionally features pre-recorded model graphs for users to analyze without the need for any installation. A free extension for VS Code can be found in both the Visual Studio Marketplace and Open VSX. For local capture, users must have Python installed along with the necessary model dependencies and sample inputs. There are also standalone coding-agent plugins that are currently in preview for Apple Silicon macOS; MCP agents can establish connections with the extension as well. While JAX/Flax graphs utilize CPU tracing to infer shapes, it is important to note that source edits for PyTorch models are not applicable in this context. By offering a seamless experience for both frameworks, tensorViz enhances the development and debugging process for users.
API Access
Has API
Yes
API Access
Has API
Yes
Screenshots View All
No images available
Integrations
Claude Code
No
Codex CLI
No
Cursor
No
GitHub
No
Lightning AI
Yes
Model Context Protocol (MCP)
No
PyTorch
Yes
T3 Code
No
Visual Studio Code
No
Integrations
Claude Code
Yes
Codex CLI
Yes
Cursor
Yes
GitHub
Yes
Lightning AI
No
Model Context Protocol (MCP)
Yes
PyTorch
No
T3 Code
Yes
Visual Studio Code
Yes
Pricing Details
Free
Free Trial
No
Free Version
Yes
Pricing Details
$0
The VS Code extension is free to install. Core graph exploration and source-edit forms work without an agent or provider API key; connecting a coding agent is optional.
Free Trial
No
Free Version
Yes
Deployment
Web-Based
Yes
On-Premises
No
iPhone App
No
iPad App
No
Android App
No
Windows
No
Mac
No
Linux
No
Chromebook
No
Deployment
Web-Based
Yes
On-Premises
No
iPhone App
No
iPad App
No
Android App
No
Windows
Yes
Mac
Yes
Linux
Yes
Chromebook
No
Customer Support
Business Hours
No
Live Rep (24/7)
No
Online Support
Yes
Customer Support
Business Hours
No
Live Rep (24/7)
No
Online Support
Yes
Types of Training
Training Docs
Yes
Webinars
No
Live Training (Online)
No
In Person
No
Types of Training
Training Docs
Yes
Webinars
No
Live Training (Online)
No
In Person
No
Vendor Details
Company Name
TorchMetrics
Country
United States
Website
torchmetrics.readthedocs.io/en/stable/
Vendor Details
Company Name
tensorViz
Website
tensorviz.dev/
Product Features
Application Development
Access Controls/Permissions
No
Code Assistance
No
Code Refactoring
No
Collaboration Tools
No
Compatibility Testing
No
Data Modeling
No
Debugging
No
Deployment Management
No
Graphical User Interface
No
Mobile Development
No
No-Code
No
Reporting/Analytics
No
Software Development
No
Source Control
No
Testing Management
No
Version Control
No
Web App Development
No
Product Features
Deep Learning
Convolutional Neural Networks
No
Document Classification
No
Image Segmentation
No
ML Algorithm Library
No
Model Training
No
Neural Network Modeling
No
Self-Learning
No
Visualization
No
Machine Learning
Deep Learning
No
ML Algorithm Library
No
Model Training
No
Natural Language Processing (NLP)
No
Predictive Modeling
No
Statistical / Mathematical Tools
No
Templates
No
Visualization
No