What problem does it solve?
This Skill streamlines the training process for large language models (LLMs), enabling users to efficiently train models with parameters exceeding 1 billion using Megatron-Core's advanced parallelism strategies.
Core Features & Use Cases
- Large-scale Training: Trains LLMs with parameters from 2B to 462B using advanced parallelism.
- Advanced Parallelism: Utilizes tensor, pipeline, sequence, and expert parallelism for efficient GPU utilization.
- High Efficiency: Achieves 47% Model FLOP Utilization on H100 GPUs.
- Use Case: Ideal for users who require high efficiency and GPU utilization for training LLMs >1B parameters.
Quick Start
Use the training-llms-megatron skill to train a large language model with the following command:
python train_llama3_8b_fp8.sh
- Ensure that Megatron-Core is installed and properly configured.
- Prepare the training data and vocabulary files.
- Run the script with the desired parameters, adjusting as necessary for your specific use case.