CoreWeave enhances its AI infrastructure for large-scale model training and inference by focusing on optimizing NVIDIA GPU clusters for both performance and reliability. This involves a multi-layered approach including bare-metal hardware, a dual-fabric network architecture (InfiniBand for compute, Ethernet for storage), topology-aware scheduling with automated node eviction and job rescheduling via SUNK (Slurm on Kubernetes), and optimized asynchronous checkpointing using Tensorizer. The SUNK system is further optimized with ART-Optimized Megatron (AOM), a new backend for the ART training library that achieves 12X higher training step throughput than previous backends by implementing shared prefix training, maximizing packed sequence length, and tuning parallelism configurations. This shared prefix approach, implemented using PyTorch's FlexAttention and custom GDN layer modifications, significantly increases trajectory density and reduces compute, memory, and memory bandwidth requirements per rollout, bringing performance closer to theoretical bounds.
All Coreweave trails