Transformer Engine

NVIDIA Transformer Engine 加速生物基礎模型 MoE 訓練：吞吐量提升達 2.21 倍

Accelerating MoE Training for Biological Foundation Models with NVIDIA Transformer Engine

本文介紹如何利用 NVIDIA BioNeMo 與 Transformer Engine 的優化算子，克服混合專家模型（MoE）在生物學應用中的訓練瓶頸，將訓練吞吐量提升至 2.21 倍。