Skip to content
Apixo
Blog
news· 1 min read· via AWS Machine Learning

Distributed training with Ray Train DLCs and EFA on EKS | Amazon Web Services

The Ray Train Deep Learning Container ships the full CUDA-through-EFA stack, so you build no image. This post walks through a reproducible sample that fine-tunes Qwen2.5-1.5B with PyTorch FSDP across eight GPUs on two EFA-enabled g6.12xlarge instances on Amazon EKS, using the…

Distributed training with Ray Train DLCs and EFA on EKS | Amazon Web Services

The Ray Train Deep Learning Container ships the full CUDA-through-EFA stack, so you build no image. This post walks through a reproducible sample that fine-tunes Qwen2.5-1.5B with PyTorch FSDP across eight GPUs on two EFA-enabled g6.12xlarge instances on Amazon EKS, using the container exactly as published.

📰 Read the full story on AWS Machine Learning: Distributed training with Ray Train DLCs and EFA on EKS | Amazon Web Services

Read the full story on AWS Machine Learning ↗
#ai-news
Try it with your own tools

One key for Claude, GPT, GLM, DeepSeek and more. Pay per token with crypto.

Get your API key

Keep reading