From the source
Lead story
Ask your AI
Top stories
Models & availability
Latest
Lead story
Top stories
Models & availability
Latest
From the source
NVRx enables async checkpointing and fault recovery for distributed GPU training on Amazon EKS.
Amazon's guide demonstrates how to integrate NVIDIA Resiliency Extension (NVRx) into PyTorch FSDP training on Amazon EKS to achieve fault-tolerant distributed training.
It covers async checkpointing, in-process restart, and in-job restart with ft_launcher, and includes benchmark results on H100 GPUs at 2-node to 8-node scale.
From the source
In this post, we show how to integrate NVIDIA Resiliency Extension (NVRx) into PyTorch Fully Sharded Data Parallel (FSDP) training on Amazon Elastic Kubernetes Service (Amazon EKS) to solve both problems.
aws.amazon.com