Skip to content
Models

Distributed training with Ray Train DLCs and EFA on EKS.

AWS Machine Learning BlogCompany announcement·
AI brief

AWS described a reproducible sample for fine-tuning the Qwen2.5-1.5B model with PyTorch FSDP across eight GPUs on two EFA-enabled g6.12xlarge instances on Amazon EKS, using Ray Train Deep Learning Containers that ship the full CUDA-through-EFA stack.

Why it matters: It shows a setup for distributed model fine-tuning on AWS that avoids building a custom container image.

Written by AI from AWS Machine Learning Blog's published text.

Source

“The Ray Train Deep Learning Container ships the full CUDA-through-EFA stack, so you build no image.”
AWS Machine Learning Blog, 28 Oct 2026, 18:19.

Published by the organisation it describes; treat claims as first-party statements.

Read original story ↗