This feature thread tracks the evolution of understanding and deploying different AI model architectures, specifically comparing Dense models with Mixture-of-Experts (MoE) models. Initial efforts focused on the fundamental differences in parameter activation and their impact on performance. This post details the introduction of advanced fused MLP kernels for MoE models, custom-built with the NVIDIA CuTe DSL. These kernels address memory and synchronization bottlenecks, enabling sync-free MoE execution. Subsequent developments have introduced TE's GroupedLinear for reduced kernel launch overhead, MXFP8 precision for memory reduction, and a fused GroupedMLP kernel that combines quantization, SwiGLU activation, and routing-weight scaling, significantly improving training throughput for MoE-based biological foundation models.
All NVIDIA trails