nemo

Scale LLM and MoE training across NVIDIA GPUs with 5D parallelism.

1|Updated Jun 4, 2026
One-click install
npx skills add https://github.com/hung-phan/ml-skills --skill nemo
Or copy as Structured Prompt for Agent
Please help me install this Agent Skill.
Skill: nemo
Source: https://github.com/hung-phan/ml-skills/tree/main/skills/ml-review/references/ml-libraries/nemo
Command: npx skills add https://github.com/hung-phan/ml-skills --skill nemo

SYSTEM DOCUMENTATION & REQUIREMENTS

💡 This Skill requires megatron-core, transformer-engine, nvidia-docker, and includes scripts (resource) and references (resource) and assets (resource) components.

What problem does it solve?

This Skill addresses the challenge of training large language models (LLMs) and mixture-of-experts (MoE) models at scale, specifically for users of the NVIDIA stack, requiring multi-node GPU clusters and advanced parallelism techniques.

Core Features & Use Cases

  • Multi-node GPU Training: Optimizes training of LLMs and MoE models across multiple GPUs and nodes.
  • Native NVIDIA Stack Support: Leverages Megatron-Core for 5D parallelism, FP8/MXFP8, and MoE Parallel Folding.
  • Verified Configurations: Provides production-ready configurations for Llama, Qwen, and DeepSeek-V3 models.
  • Deployment Path: Streamlines the process from training to deployment with TensorRT-LLM, vLLM, and NIM.
  • Use Case: Ideal for scaling pretraining or post-training across many NVIDIA GPUs, training MoE models, and deploying to NIM/TRT-LLM.

Quick Start

Use the nemo skill to train a large LLM model on a multi-node GPU cluster.

Frequently Asked Questions about nemo

High-intent search queries and answers about installing and using this skill.

FAQPage Schema
How do I scale LLM and MoE model training across multiple NVIDIA GPUs and nodes?

To scale LLM and MoE training across multiple NVIDIA GPUs, you can use this Skill to apply 5D parallelism and MoE Parallel Folding across multi-node clusters. It leverages Megatron-Core to optimize distributed training workloads natively within the NVIDIA stack.

Does this Skill support training configurations for Llama, Qwen, and DeepSeek-V3 models?

Yes, this Skill provides verified, production-ready training configurations for Llama, Qwen, and DeepSeek-V3 models. These built-in configurations ensure proper setup for pretraining and post-training across multi-node GPU clusters.

What is the best way to deploy trained models to NIM or TensorRT-LLM?

The best way to deploy trained models to NIM or TensorRT-LLM is using this Skill's built-in deployment path. It streamlines the transition from multi-node GPU training directly to inference using TensorRT-LLM, vLLM, and NIM frameworks.

Can I use FP8 and MXFP8 precision for large language model training?

Yes, you can use FP8 and MXFP8 precision for large language model training. The Skill natively supports these precision formats through the transformer-engine dependency to optimize memory and compute efficiency during multi-node training.

Do I need nvidia-docker and Megatron-Core to run multi-node LLM training?

Yes, you need nvidia-docker and Megatron-Core dependencies to run multi-node LLM training. These components are required to manage the NVIDIA container environment and execute the 5D parallelism techniques used for scaling.

Why use MoE Parallel Folding when training mixture-of-experts models on GPU clusters?

MoE Parallel Folding is used when training mixture-of-experts models on GPU clusters to optimize expert distribution and reduce communication overhead. This technique enables efficient scaling of MoE architectures across multiple nodes using the NVIDIA stack.