Amazon Unveils Trainium2 AI Chip and Collaborates with Nvidia to Expand Cloud Offerings

The move addresses the surging demand for GPUs following the success of AI technologies like OpenAI's ChatGPT, resulting in supply shortages in the market.

Amazon Unveils Trainium2 AI Chip and Collaborates with Nvidia to Expand Cloud Offerings
Image / AWS Reinvent

Amazon Web Services (AWS) has made significant strides in the AI domain, introducing Trainium2, a powerful chip designed for training artificial intelligence models, and expanding its collaboration with Nvidia to offer access to the latest H200 Tensor Core graphics processing units.

During the recent Reinvent conference in Las Vegas, AWS unveiled its strategy to empower customers with cutting-edge AI capabilities through a multi-faceted approach. The announcement encompasses Amazon's proprietary Trainium2 chip, the upcoming availability of Nvidia's H200 GPUs, and the launch of Graviton4 processors for general-purpose computing.

Notably, AWS customers can already initiate testing on the new general-purpose Graviton4 chips, aimed at providing enhanced performance and efficiency compared to their predecessors.

In a bid to stand out among cloud providers, AWS is diversifying its offerings by integrating high-performance solutions, including Nvidia's sought-after GPUs, known for their pivotal role in AI applications. The move addresses the surging demand for GPUs following the success of AI technologies like OpenAI's ChatGPT, resulting in supply shortages in the market.

Amazon's comprehensive approach, combining the development of its AI chips with access to Nvidia's latest offerings, positions the cloud giant against competitors like Microsoft. This strategy aligns with Microsoft's recent introduction of its own AI chip, the Maia 100, and plans to leverage Nvidia's H200 GPUs in the Azure cloud.

The H200 GPU from Nvidia, an upgrade from its predecessor H100, promises nearly double the performance output and caters to the escalating demand from various industries, including large-scale language model training.

AWS's Trainium2 chips, tailored for training AI models, are poised to drive advancements in AI chatbots and other AI-powered applications. Amazon boasts a fourfold performance improvement over the original model, attracting interest from entities like startup Databricks and Amazon-backed Anthropic.

The Graviton4 processors, built on Arm architecture, emphasize energy efficiency and tout a 30% performance enhancement compared to previous iterations. This advancement could potentially offer organizations improved output at a better value, especially amid economic fluctuations and rising operational costs.

As part of its extended partnership with Nvidia, AWS is set to deploy over 16,000 Nvidia GH200 Grace Hopper Superchips. This infrastructure, equipped with Nvidia GPUs and Arm-based general-purpose processors, will be available for Nvidia's R&D endeavors and AWS customers' utilization.

Despite not disclosing specific release dates, AWS plans to offer instances with Nvidia H200 chips and Trainium2 silicon shortly. Meanwhile, customers can commence testing Graviton4 virtual-machine instances, expected to be commercially available in the coming months.

AWS's continuous investment in the Graviton and Trainium programs underscores Amazon's commitment to meeting evolving market demands and advancing AI capabilities within its cloud ecosystem.