콘텐츠로 이동

Distributed Training

This tutorial guides you through distributed training of large language models on AWS Trainium clusters.


Overview

AWS Trainium is a custom chip optimized for deep learning training, delivering the best cost-performance ratio. The neuronx-distributed library enables easy implementation of data/tensor/pipeline parallel training.


Prerequisites

  • 2+ trn1.32xlarge instances
  • EFA (Elastic Fabric Adapter) networking enabled
  • Neuron SDK 2.18 or higher

Step 1: Cluster Configuration

Coming Soon

Detailed step-by-step guide will be added soon.


References