Instructions to use KexuanShi/Megatron-LM with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- NeMo
How to use KexuanShi/Megatron-LM with NeMo:
# tag did not correspond to a valid NeMo domain.
- Notebooks
- Google Colab
- Kaggle
| # Megatron Core User Guide | |
| **Megatron Core** is a GPU-optimized library for training large language models at scale. It provides modular, composable building blocks for creating custom training frameworks with state-of-the-art parallelism strategies and performance optimizations. | |
| Megatron Core offers a flexible, reusable foundation for building large-scale transformer training systems. **Megatron-LM** serves as a reference implementation demonstrating how to use Megatron Core components to train models with billions to trillions of parameters across distributed GPU clusters. | |
| ## Key Features | |
| * Composable transformer building blocks (attention, MLP, etc.) | |
| * Advanced parallelism strategies (TP, PP, DP, EP, CP) | |
| * Pipeline schedules and distributed optimizers | |
| * Mixed precision support (FP16, BF16, FP8) | |
| * GPU-optimized kernels and memory management | |
| * High-performance dataloaders and dataset utilities | |
| * Model architectures (LLaMA, Qwen, DeepSeek, GPT, Mamba, etc.) | |
| ```{toctree} | |
| :maxdepth: 2 | |
| :hidden: | |
| :caption: Get Started | |
| get-started/quickstart | |
| ``` | |
| ```{toctree} | |
| :maxdepth: 2 | |
| :hidden: | |
| :caption: Basic Usage | |
| user-guide/data-preparation | |
| user-guide/training-examples | |
| user-guide/parallelism-guide | |
| ``` | |
| ```{toctree} | |
| :maxdepth: 2 | |
| :hidden: | |
| :caption: Supported Models | |
| models/index | |
| ``` | |
| ```{toctree} | |
| :maxdepth: 2 | |
| :hidden: | |
| :caption: Advanced Features | |
| user-guide/features/moe | |
| user-guide/features/context_parallel | |
| user-guide/features/custom_fsdp | |
| user-guide/features/dist_optimizer | |
| user-guide/features/optimizer_cpu_offload | |
| user-guide/features/pipeline_parallel_layout | |
| user-guide/features/megatron_energon | |
| user-guide/features/megatron_rl | |
| user-guide/features/tokenizers | |
| ``` | |
| ```{toctree} | |
| :maxdepth: 1 | |
| :hidden: | |
| :caption: Developer Guide | |
| developer/contribute | |
| developer/submit | |
| developer/oncall | |
| developer/generate_docs | |
| ``` | |
| ```{toctree} | |
| :maxdepth: 2 | |
| :hidden: | |
| :caption: Discussions | |
| advanced/index | |
| ``` | |
| ```{toctree} | |
| :maxdepth: 2 | |
| :hidden: | |
| :caption: API Reference | |
| api-guide/index | |
| apidocs/index.rst | |
| ``` |