Average Ratings 0 Ratings
Average Ratings 0 Ratings
Description
It enables efficient training on Amazon Elastic Compute Cloud (Amazon EC2) Trn1 instances powered by AWS Trainium. Additionally, for model deployment, it facilitates both high-performance and low-latency inference utilizing AWS Inferentia-based Amazon EC2 Inf1 instances along with AWS Inferentia2-based Amazon EC2 Inf2 instances. With the Neuron SDK, users can leverage widely-used frameworks like TensorFlow and PyTorch to effectively train and deploy machine learning (ML) models on Amazon EC2 Trn1, Inf1, and Inf2 instances with minimal alterations to their code and no reliance on vendor-specific tools. The integration of the AWS Neuron SDK with these frameworks allows for seamless continuation of existing workflows, requiring only minor code adjustments to get started. For those involved in distributed model training, the Neuron SDK also accommodates libraries such as Megatron-LM and PyTorch Fully Sharded Data Parallel (FSDP), enhancing its versatility and scalability for various ML tasks. By providing robust support for these frameworks and libraries, it significantly streamlines the process of developing and deploying advanced machine learning solutions.
Description
Amazon EC2 Trn2 instances, equipped with AWS Trainium2 chips, are specifically designed to deliver exceptional performance in the training of generative AI models, such as large language and diffusion models. Users can experience cost savings of up to 50% in training expenses compared to other Amazon EC2 instances. These Trn2 instances can accommodate as many as 16 Trainium2 accelerators, boasting an impressive compute power of up to 3 petaflops using FP16/BF16 and 512 GB of high-bandwidth memory. For enhanced data and model parallelism, they are built with NeuronLink, a high-speed, nonblocking interconnect, and offer a substantial network bandwidth of up to 1600 Gbps via the second-generation Elastic Fabric Adapter (EFAv2). Trn2 instances are part of EC2 UltraClusters, which allow for scaling up to 30,000 interconnected Trainium2 chips within a nonblocking petabit-scale network, achieving a remarkable 6 exaflops of compute capability. Additionally, the AWS Neuron SDK provides seamless integration with widely used machine learning frameworks, including PyTorch and TensorFlow, making these instances a powerful choice for developers and researchers alike. This combination of cutting-edge technology and cost efficiency positions Trn2 instances as a leading option in the realm of high-performance deep learning.
API Access
Has API
No
API Access
Has API
No
Integrations
AWS Deep Learning AMIs
Yes
AWS Trainium
Yes
Amazon EC2 Capacity Blocks for ML
Yes
Amazon EC2 G5 Instances
Yes
Amazon EC2 Inf1 Instances
Yes
Amazon EC2 P4 Instances
Yes
Amazon EC2 P5 Instances
Yes
Amazon EC2 Trn1 Instances
Yes
Amazon EC2 UltraClusters
Yes
Amazon EKS
Yes
Integrations
AWS Deep Learning AMIs
Yes
AWS Trainium
Yes
Amazon EC2 Capacity Blocks for ML
Yes
Amazon EC2 G5 Instances
Yes
Amazon EC2 Inf1 Instances
Yes
Amazon EC2 P4 Instances
Yes
Amazon EC2 P5 Instances
Yes
Amazon EC2 Trn1 Instances
Yes
Amazon EC2 UltraClusters
Yes
Amazon EKS
Yes
Pricing Details
No price information available.
Free Trial
No
Free Version
No
Pricing Details
No price information available.
Free Trial
No
Free Version
No
Deployment
Web-Based
Yes
On-Premises
No
iPhone App
No
iPad App
No
Android App
No
Windows
No
Mac
No
Linux
No
Chromebook
No
Deployment
Web-Based
Yes
On-Premises
No
iPhone App
No
iPad App
No
Android App
No
Windows
No
Mac
No
Linux
No
Chromebook
No
Customer Support
Business Hours
Yes
Live Rep (24/7)
No
Online Support
Yes
Customer Support
Business Hours
Yes
Live Rep (24/7)
Yes
Online Support
Yes
Types of Training
Training Docs
Yes
Webinars
No
Live Training (Online)
No
In Person
No
Types of Training
Training Docs
Yes
Webinars
Yes
Live Training (Online)
No
In Person
Yes
Vendor Details
Company Name
Amazon Web Services
Founded
2006
Country
United States
Website
aws.amazon.com/machine-learning/neuron/
Vendor Details
Company Name
Amazon
Founded
1994
Country
United States
Website
aws.amazon.com/ec2/instance-types/trn2/
Product Features
Deep Learning
Convolutional Neural Networks
No
Document Classification
No
Image Segmentation
No
ML Algorithm Library
No
Model Training
No
Neural Network Modeling
No
Self-Learning
No
Visualization
No
Machine Learning
Deep Learning
No
ML Algorithm Library
No
Model Training
No
Natural Language Processing (NLP)
No
Predictive Modeling
No
Statistical / Mathematical Tools
No
Templates
No
Visualization
No
Product Features
Deep Learning
Convolutional Neural Networks
No
Document Classification
No
Image Segmentation
No
ML Algorithm Library
No
Model Training
No
Neural Network Modeling
No
Self-Learning
No
Visualization
No
Machine Learning
Deep Learning
No
ML Algorithm Library
No
Model Training
No
Natural Language Processing (NLP)
No
Predictive Modeling
No
Statistical / Mathematical Tools
No
Templates
No
Visualization
No