+
+

Related Products

  • Gemini Enterprise Agent Platform
    1,161 Ratings
    Visit Website
  • Runpod
    230 Ratings
    Visit Website
  • Compute with Hivenet
    4 Ratings
    Visit Website
  • IONOS Cloud GPU Servers
    45,340 Ratings
    Visit Website
  • Google AI Studio
    41 Ratings
    Visit Website
  • Fraud.net
    56 Ratings
    Visit Website
  • Qloo
    23 Ratings
    Visit Website
  • Google Cloud BigQuery
    2,027 Ratings
    Visit Website
  • LM-Kit.NET
    29 Ratings
    Visit Website
  • LTX
    182 Ratings
    Visit Website

About

Amazon Elastic Compute Cloud (EC2) Trn1 instances, powered by AWS Trainium chips, are purpose-built for high-performance deep learning training of generative AI models, including large language models and latent diffusion models. Trn1 instances offer up to 50% cost-to-train savings over other comparable Amazon EC2 instances. You can use Trn1 instances to train 100B+ parameter DL and generative AI models across a broad set of applications, such as text summarization, code generation, question answering, image and video generation, recommendation, and fraud detection. The AWS Neuron SDK helps developers train models on AWS Trainium (and deploy models on the AWS Inferentia chips). It integrates natively with frameworks such as PyTorch and TensorFlow so that you can continue using your existing code and workflows to train models on Trn1 instances.

About

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.

Platforms Supported

Windows Not Supported
Mac Not Supported
Linux Not Supported
Cloud Supported
On-Premises Not Supported
iPhone Not Supported
iPad Not Supported
Android Not Supported
Chromebook Not Supported

Platforms Supported

Windows Supported
Mac Supported
Linux Supported
Cloud Supported
On-Premises Not Supported
iPhone Not Supported
iPad Not Supported
Android Not Supported
Chromebook Not Supported

Audience

Organizations and developers requiring a solution for training large-scale deep learning and generative AI models

Audience

Machine learning engineers, PyTorch developers, researchers and students exploring model architecture, tensor shapes and architecture changes.

Support

Phone Support Supported
24/7 Live Support Supported
Online Supported

Support

Phone Support Not Supported
24/7 Live Support Not Supported
Online Supported

API

Offers API Not Supported

API

Offers API Supported

Screenshots and Videos

Screenshots and Videos

No images available

Pricing

$1.34 per hour
Free Version Not Supported
Free Trial Not Supported

Pricing

$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 Version Supported
Free Trial Not Supported

Reviews/Ratings

Overall 0.0 / 5
ease 0.0 / 5
features 0.0 / 5
design 0.0 / 5
support 0.0 / 5

This software hasn't been reviewed yet. Be the first to provide a review:

Review this Software

Reviews/Ratings

Overall 0.0 / 5
ease 0.0 / 5
features 0.0 / 5
design 0.0 / 5
support 0.0 / 5

This software hasn't been reviewed yet. Be the first to provide a review:

Review this Software

Training

Documentation Supported
Webinars Supported
Live Online Not Supported
In Person Supported

Training

Documentation Supported
Webinars Not Supported
Live Online Not Supported
In Person Not Supported

Company Information

Amazon
Founded: 1994
United States
aws.amazon.com/ec2/instance-types/trn1/

Company Information

tensorViz
tensorviz.dev/

Alternatives

Alternatives

AWS Neuron

AWS Neuron

Amazon Web Services
AWS Neuron

AWS Neuron

Amazon Web Services
AWS Trainium

AWS Trainium

Amazon Web Services

Categories

AI Fine-Tuning Supported
AI Infrastructure Supported
Deep Learning Supported
Machine Learning Supported

Categories

Deep Learning Supported
Machine Learning Supported

Integrations

AWS Deep Learning AMIs Supported
AWS Deep Learning Containers Supported
AWS Inferentia Supported
AWS Neuron Supported
AWS Nitro System Supported
AWS Trainium Supported
Amazon EC2 Supported
Amazon EC2 Capacity Blocks for ML Supported
Amazon EC2 G5 Instances Supported
Amazon EC2 P4 Instances Supported
Amazon EC2 P5 Instances Supported
Amazon Web Services (AWS) Supported
Claude Code Not Supported
Codex CLI Not Supported
GitHub Not Supported
Model Context Protocol (MCP) Not Supported
PyTorch Supported
T3 Code Not Supported
TensorFlow Supported
Visual Studio Code Not Supported

Integrations

AWS Deep Learning AMIs Not Supported
AWS Deep Learning Containers Not Supported
AWS Inferentia Not Supported
AWS Neuron Not Supported
AWS Nitro System Not Supported
AWS Trainium Not Supported
Amazon EC2 Not Supported
Amazon EC2 Capacity Blocks for ML Not Supported
Amazon EC2 G5 Instances Not Supported
Amazon EC2 P4 Instances Not Supported
Amazon EC2 P5 Instances Not Supported
Amazon Web Services (AWS) Not Supported
Claude Code Supported
Codex CLI Supported
GitHub Supported
Model Context Protocol (MCP) Supported
PyTorch Not Supported
T3 Code Supported
TensorFlow Not Supported
Visual Studio Code Supported
Claim Amazon EC2 Trn1 Instances and update features and information
Claim Amazon EC2 Trn1 Instances and update features and information
Claim tensorViz and update features and information
Claim tensorViz and update features and information