https://duoan.github.io/tags/megatron-lm/ Recent content in Megatron-LM on Duo's Tech Blog https://duoan.github.io/images/papermod-cover.png https://duoan.github.io/images/papermod-cover.png Hugo -- 0.153.1 en-us Sun, 18 May 2025 00:00:00 +0000 https://duoan.github.io/posts/tensor-parallelism-megatron/ Sun, 18 May 2025 00:00:00 +0000 https://duoan.github.io/posts/tensor-parallelism-megatron/ <h1 id="tensor-parallelism-in-megatron-lm-splitting-layers-not-stacks">Tensor Parallelism in Megatron-LM: Splitting Layers, Not Stacks</h1> <p>Pipeline parallelism splits a model by depth. Tensor parallelism splits the math inside one layer.</p> <p>Megatron-LM made tensor parallelism practical for Transformers by choosing split points that preserve local GEMMs and put collectives only where the algebra requires them. The original Megatron paper introduced the core intra-layer pattern (<a href="https://arxiv.org/abs/1909.08053">arXiv:1909.08053</a>). The later Megatron-LM systems paper showed how that TP axis composes with data and pipeline parallelism at cluster scale (<a href="https://arxiv.org/abs/2104.04473">arXiv:2104.04473</a>).</p> https://duoan.github.io/posts/llm-training-series/ Sun, 02 Mar 2025 00:00:00 +0000 https://duoan.github.io/posts/llm-training-series/ <h1 id="llm-training-series-a-systems-map-for-distributed-transformers">LLM Training Series: A Systems Map for Distributed Transformers</h1> <p>LLM training stops being &ldquo;run the model on more GPUs&rdquo; the moment one replica no longer fits, one link becomes the step time, or one schedule leaves half the cluster idle. From that point on, the training run is a placement problem.</p> <p>The stable question is simple: <strong>which bytes are replicated, which bytes are sharded, and which link moves them on the critical path?</strong> Data parallelism, tensor parallelism, pipeline parallelism, ZeRO, sequence/context parallelism, and expert parallelism are different answers to that question. The public systems that matter - GPipe, Megatron-LM, DeepSpeed ZeRO, PyTorch DDP, GShard, Switch Transformer, and modern Megatron-Core - all expose the same constraints in different shapes (<a href="https://arxiv.org/abs/1811.06965">GPipe</a>, <a href="https://arxiv.org/abs/1909.08053">Megatron-LM</a>, <a href="https://arxiv.org/abs/1910.02054">ZeRO</a>, <a href="https://arxiv.org/abs/2006.15704">PyTorch Distributed</a>, <a href="https://arxiv.org/abs/2006.16668">GShard</a>, <a href="https://arxiv.org/abs/2101.03961">Switch Transformer</a>).</p>