What is tensorViz?

tensorViz offers a unique platform that allows developers to explore PyTorch models and JAX/Flax programs through interactive graph visualizations. For PyTorch, users can delve into nested modules, inspect tensor shapes captured during a single forward pass, navigate to the related Python source code, and witness architectural changes along with code variations. The platform also includes pre-recorded model graphs that can be analyzed without any installation requirements. A free extension compatible with VS Code is available on both the Visual Studio Marketplace and Open VSX. To enable local capture, users need to have Python installed along with the relevant model dependencies and sample inputs. Additionally, there are standalone coding-agent plugins in preview specifically designed for Apple Silicon macOS, allowing MCP agents to connect seamlessly with the extension. While JAX/Flax graphs rely on CPU tracing to determine shapes, it's crucial to understand that source modifications for PyTorch models do not apply in this scenario. By providing a cohesive experience for both frameworks, tensorViz significantly improves the development and debugging workflow for its users, ultimately leading to more efficient coding practices. This dual framework support ensures that developers have the tools they need to effectively manage their projects and enhance their productivity.

Pricing

Price Starts At:
$0
Price Overview:
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.

Integrations

Offers API?:
Yes, tensorViz provides an API

Screenshots and Video

Get Started

Company Facts

Company Name:
tensorViz
Company Website:
tensorviz.dev/

Product Details

Deployment
SaaS
Windows
Mac
Linux
Training Options
Documentation Hub
Video Library
Support
Web-Based Support

Product Details

Target Company Sizes
Individual
1-10
11-50
51-200
201-500
501-1000
1001-5000
5001-10000
10001+
Target Organization Types
Mid Size Business
Small Business
Enterprise
Freelance
Nonprofit
Government
Startup
Supported Languages
English

tensorViz Categories and Features