NVIDIA's Transformer Engine Boosts MoE Training in JAX by 10x
NVIDIA's Transformer Engine accelerates Dropless Mixture-of-Experts (MoE) training in JAX, achieving a 10x performance gain and 97% scaling efficiency.
Markets