World's Fair 2024
Accelerating Mixture of Experts Training With Rail Optimized InfiniBand Networking in Crusoe Cloud
Overview
This talk focuses on optimizing the infrastructure for training large AI models, specifically addressing the bottlenecks in distributed training caused by network communication. Crusoe Cloud, an AI cloud platform powered by renewable energy, highlights its rail-optimized InfiniBand networking solution designed to accelerate Mixture of Experts (MoE) training by reducing the time gpus spend idle waiting for data exchange.
Who should watch
- AI Engineers and ML Engineers working with distributed training.
- Infrastructure providers and cloud engineers focused on high-performance computing for AI.
- Product Managers and builders evaluating AI cloud platforms for performance and efficiency.
- Anyone interested in reducing model training time and costs.
Key takeaways
- Network communication can account for 25-30% of distributed training time, during which gpus are idle.
- Crusoe Cloud's platform is built on three pillars: high performance, ease of use for AI engineers, and climate alignment using renewable energy.
- The platform offers VMs with attached gpus, CPU instances, persistent storage, and traditional VPC networking.
- A key innovation is their rail-optimized InfiniBand cluster networking, which improves GPU-to-GPU communication.
- Nvidia's NCCL PXN feature allows internal host switches to facilitate cross-rail communication, reducing latency and improving bandwidth utilization.
- Testing with the open-source Mixi model (a sparse MoE) showed a 14% improvement in training time when using Crusoe's optimized networking.
- This performance improvement directly translates to reduced model training costs and faster iteration cycles for AI developers.
Notable quotes
*The networking part takes without any optimization about 25 30% of the time of the training time.*
*14% of improvement is something that can be directly connected to the time to train the model that can be directly connected to cost of training the model.*
Unofficial community note. Prefer the recording for nuance.