AWS Neuron

AWS Neuron

Amazon Web Services
+
+

Related Products

  • Gemini Enterprise Agent Platform
    1,161 Ratings
    Visit Website
  • Compute with Hivenet
    4 Ratings
    Visit Website
  • Runpod
    230 Ratings
    Visit Website
  • Anaconda
    625 Ratings
    Visit Website
  • IONOS Cloud GPU Servers
    45,340 Ratings
    Visit Website
  • Elevate EDI
    12 Ratings
    Visit Website
  • Servers.com by Nexcess
    15 Ratings
    Visit Website
  • Qloo
    23 Ratings
    Visit Website
  • KrakenD
    76 Ratings
    Visit Website
  • Fraud.net
    56 Ratings
    Visit Website

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 Videos

No 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/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 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 Information

Amazon Web Services
Founded: 2006
United States
aws.amazon.com/machine-learning/neuron/

Company Information

tensorViz
tensorviz.dev/

Alternatives

Alternatives

AWS Neuron

AWS Neuron

Amazon Web Services

Categories

AI Inference 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 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
Amazon EKS Supported
Amazon Elastic Container Service (Amazon ECS) Supported
Amazon SageMaker Supported
Amazon Web Services (AWS) Supported
Claude Code Not Supported
Codex CLI Not Supported
Cursor Not Supported
GitHub Not Supported
Model Context Protocol (MCP) Not Supported
T3 Code Not 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
Amazon EKS Not Supported
Amazon Elastic Container Service (Amazon ECS) Not Supported
Amazon SageMaker Not Supported
Amazon Web Services (AWS) Not Supported
Claude Code Supported
Codex CLI Supported
Cursor Supported
GitHub Supported
Model Context Protocol (MCP) Supported
T3 Code Supported
Claim AWS Neuron and update features and information
Claim AWS Neuron and update features and information
Claim tensorViz and update features and information
Claim tensorViz and update features and information