Megablocks MoE

Databricks open library for efficient Mixture-of-Experts training at scale.