This document is relevant for: Inf1, Inf2, Trn1, Trn2

AWS AI Chips Architecture

AWS AI Chips (Neuron Devices) are the accelerated machine learning chips (e.g. Inferentia or Trainium) that enable Trn and Inf instance.

For a detailed description of current Neuron Devices:

• Trainium2 architecture

• Trainium Architecture

• Inferentia2 Architecture

• Inferentia Architecture

This document is relevant for: Inf1, Inf2, Trn1, Trn2