SageMaker biblioteca de paralelismo de modelos v2

A biblioteca de paralelismo de modelos de SageMaker IA da Amazon é um recurso de SageMaker IA que permite treinamento otimizado e de alto desempenho em grande escala em instâncias de computação aceleradas de SageMaker IA. O Principais características da biblioteca de paralelismo de SageMaker modelos v2 inclui técnicas e otimizações para acelerar e simplificar o treinamento de grandes modelos, como paralelismo híbrido de dados fragmentados, paralelismo de tensores, pontos de verificação de ativação e descarregamento de ativação. Você pode usar a biblioteca SMP para acelerar o treinamento e o ajuste fino de grandes modelos de linguagem (LLMs), modelos de visão amplos (LVMs) e modelos básicos (FMs) com centenas de bilhões de parâmetros.

A biblioteca de paralelismo de SageMaker modelos v2 (SMP v2) alinha a biblioteca APIs e os métodos com o paralelismo de dados PyTorch totalmente fragmentado (FSDP) de código aberto, o que oferece o benefício das otimizações de desempenho do SMP com o mínimo de alterações no código. Com o SMP v2, você pode melhorar o desempenho computacional do treinamento de um state-of-the-art grande modelo em SageMaker IA trazendo seus scripts de treinamento do PyTorch FSDP para a IA. SageMaker

Você pode usar o SMP v2 para trabalhos gerais de SageMaker treinamento e cargas de trabalho de treinamento distribuídas em clusters. Amazon SageMaker HyperPod