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.