Audience
Machine learning engineers, PyTorch developers, researchers and students exploring model architecture, tensor shapes and architecture changes.
About tensorViz
tensorViz helps developers explore PyTorch models and JAX/Flax programs as interactive graphs. For PyTorch, users can expand nested modules, inspect tensor shapes captured from one forward pass, jump to Python source, and review architecture changes beside code diffs. The website offers recorded model graphs to explore without installation. The free VS Code extension is available from the Visual Studio Marketplace and Open VSX. Local capture requires Python, model dependencies, and example inputs. Standalone coding-agent plugins are an Apple Silicon macOS preview; MCP agents can also connect to the extension. JAX/Flax graphs use CPU tracing with inferred shapes, and PyTorch source edits are unavailable for them.
Other Popular Alternatives & Related Software
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 about Google Cloud Deep Learning VM Image
Muse Spark Open
Muse Spark Open is an open-weight, natively multimodal language model for developers and organizations building AI applications, coding systems, and agentic workflows. The released model has 816 billion total parameters, 42 billion active parameters, and a 1-million-token context window. Teams can download the model weights, run inference through supported engines such as vLLM, SGLang, TensorRT-LLM, Hugging Face Transformers, Apple MLX, llama.cpp, or Unsloth, and customize the model using PyTorch. It supports long-context reasoning, software engineering, coding, tool use, and multimodal inputs. Deployment options include cloud inference through listed API providers or self-hosting on compatible GPU systems; Meta documents a desktop setup using two Apple M5 Ultra Mac Studio computers. The model is released under the Muse Model License with limited commercial terms.
Learn more about Muse Spark Open
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.
Learn more about TorchMetrics
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 about PyTorch
Pricing
Starting Price:
$0
Pricing Details:
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 Version:
Free Version available.
Company Information
tensorViz
tensorviz.dev/
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
Windows
Mac
Linux
Training
Documentation
Videos
Support
Online
tensorViz Frequently Asked Questions
Q: What kinds of users and organization types does tensorViz work with?
tensorViz works with these users and organization types: Mid Size Business, Small Business, Enterprise, Freelance, Nonprofit, Government, and Startup.
Q: What languages does tensorViz support in their product?
tensorViz supports these languages: English.
Q: What kind of support options does tensorViz offer?
tensorViz offers support via online.
Q: What other applications or services does tensorViz integrate with?
tensorViz integrates with: Claude Code, Codex CLI, Cursor, GitHub, Model Context Protocol (MCP), T3 Code, and Visual Studio Code.
Q: Does tensorViz have an API?
Yes, tensorViz has an API.
Q: What type of training does tensorViz provide?
tensorViz provides training in the form of documentation and videos.
Q: How much does tensorViz cost?
Pricing for tensorViz starts at $0.
