Home/Model Training/Megatron-LM
Megatron-LM logo

Megatron-LM

NVIDIA/Megatron-LM

Ongoing research training transformer models at scale

GraphCanon updated 1w · GitHub synced 1w · 32 views this month

17k stars4.3k forksLast push 1w Python Other

Decision brief

Megatron-LM from NVIDIA is a research-focused tool for developing and training large-scale language models with transformer architectures, emphasizing efficient parallelism across multiple GPUs.

Good fit when

  • The tool is particularly beneficial when your project is GPU-centric and benefits from advanced parallelism techniques such as Tensor, Pipeline, Data, Expert, and Cluster Parallelisms (TP, PP, DP, EP,
  • and CP) provided by NVIDIA's ecosystem.

Avoid when

  • Avoid Megatron-LM if your computational setup does not include NVIDIA GPUs as it leverages GPU-specific features and parallelisms that may not be available or efficient on non-NVIDIA hardware.
  • If you need portability across various hardware without depending on proprietary optimizations, other tools might better serve your needs.
Requirements:
Min 32 GB RAM; Requires NVIDIA GPUs for optimized performance. Non-GPU usage is not supported or recommended.; Installation from source can be resource-intensive and may require limiting parallel compilation jobs to avoid running out of memory.

Observed Jul 11, 2026 · Source: enrich:decision_facts

Verify the decision

Maintenance and security

Full trust report
Maintenance
Very active (0d since push)
As of 1w
Provenance
Not a fork · Organization account
As of 1w
Security (OSV)
No lockfile
As of 1mo

Public GitHub metadata and optional OSV scans. Signals, not a guarantee. Trust methodology.

Backing

Company context for Nvidia. Display-only - separate from trust and ranking.

Company
NVIDIA Corporation·GitHub org profile·1mo
Employees
11,528·Wikidata (P1128 employees)·1mo
Commercial model
Pure OSS·GitHub org profile (public repos)·1mo

Install

pip install Megatron-LM
PyPI

How it fits your stack(9)

Typed graph edges - alternatives, integrations, successors, and dependencies. Ranked by relationship type, not raw GitHub stars.

Related

Relationship graph

Optional deeper exploration of typed edges and category neighbours.

Similar tools

Same-category neighbours not already linked as typed edges.

Evidence and technical details

Sourced facts, taxonomy, compatibility claims, README excerpt, and machine-readable endpoints.

Overview

Megatron-LM is a repository from NVIDIA focused on the development and training of large-scale language models using transformer architectures. It provides tools for efficient parallelism strategies across multiple GPUs.

Capability facts

Languages
python

Source: github.language+pyproject.toml · Aug 7, 2026

Categories

Graph entities

Compatibility

Sourced claims from the README excerpt - not unsourced marketing copy.

Python runtimePython

Source: README excerpt (regex_v1, Aug 7, 2026)

uv pip install megatron-core
Source link

Tags

README

Getting Started

Install from PyPI:

uv pip install megatron-core

Or clone and install from source:

git clone https://github.com/NVIDIA/Megatron-LM.git
cd Megatron-LM
uv pip install -e .

Note: Building from source can use a lot of memory. If the build runs out of memory, limit parallel compilation jobs by setting MAX_JOBS (for example, MAX_JOBS=4 uv pip install -e .).

For NVIDIA GPU Cloud (NGC) container setup and all installation options, review the Installation Guide.

For agents

This page has a .md twin and JSON over the API.

Was this helpful?

Anonymous feedback helps us improve pages and translations.