AWS NeuronAmazon Web Services
|
||||||
Related Products
|
||||||
About
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).
|
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 in need of an SDK solution with a compiler, runtime, and profiling tools that unlocks high-performance and cost-effective deep learning acceleration
|
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
Not 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 VideosNo images available
|
|||||
Pricing
No information available.
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/
|
Reviews/
|
|||||
Training
Documentation
Supported
Webinars
Not Supported
Live Online
Not Supported
In Person
Not Supported
|
Training
Documentation
Supported
Webinars
Not Supported
Live Online
Not Supported
In Person
Not Supported
|
|||||
Company InformationAmazon Web Services
Founded: 2006
United States
aws.amazon.com/machine-learning/neuron/
|
Company InformationtensorViz
tensorviz.dev/
|
|||||
Alternatives |
Alternatives |
|||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
Categories |
Categories |
|||||
Integrations
AWS Deep Learning AMIs
Supported
AWS Deep Learning Containers
Supported
AWS Trainium
Supported
Amazon EC2 Capacity Blocks for ML
Supported
Amazon EC2 G5 Instances
Supported
Amazon EC2 Inf1 Instances
Supported
Amazon EC2 P4 Instances
Supported
Amazon EC2 Trn1 Instances
Supported
Amazon EC2 Trn2 Instances
Supported
Amazon EC2 UltraClusters
Supported
|
Integrations
AWS Deep Learning AMIs
Not Supported
AWS Deep Learning Containers
Not Supported
AWS Trainium
Not Supported
Amazon EC2 Capacity Blocks for ML
Not Supported
Amazon EC2 G5 Instances
Not Supported
Amazon EC2 Inf1 Instances
Not Supported
Amazon EC2 P4 Instances
Not Supported
Amazon EC2 Trn1 Instances
Not Supported
Amazon EC2 Trn2 Instances
Not Supported
Amazon EC2 UltraClusters
Not Supported
|
|||||
|
|
|