distributed trainer for LLMs
☆590May 20, 2024Updated 2 years ago
Alternatives and similar repositories for Megatron-LLM
Users that are interested in Megatron-LLM are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Best practice for training LLaMA models in Megatron-LM☆666Jan 2, 2024Updated 2 years ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆1,449Mar 20, 2024Updated 2 years ago
- The official repo of Pai-Megatron-Patch for LLM & VLM large scale training developed by Alibaba Cloud.☆1,602Dec 15, 2025Updated 10 months ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆69Jul 20, 2023Updated 3 years ago
- Ongoing research training transformer language models at scale, including: BERT & GPT-2☆2,268Aug 14, 2025Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆84Sep 9, 2023Updated 3 years ago
- Ongoing research training transformer models at scale☆18,316Updated this week
- Minimalistic large language model 3D-parallelism training☆2,851Oct 8, 2026Updated 3 weeks ago
- Medusa: Simple Framework for Accelerating LLM Generation with Multiple Decoding Heads☆2,774Jun 25, 2024Updated 2 years ago
- A family of open-sourced Mixture-of-Experts (MoE) Large Language Models☆1,702Mar 8, 2024Updated 2 years ago
- Zero Bubble Pipeline Parallelism☆468May 7, 2025Updated last year
- Microsoft Automatic Mixed Precision Library☆634Dec 1, 2025Updated 11 months ago
- ☆1,600Mar 25, 2026Updated 7 months ago
- Memory optimization and training recipes to extrapolate language models' context length to 1 million tokens, with minimal hardware.☆761Sep 27, 2024Updated 2 years ago
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- A repo for distributed training of language models with Reinforcement Learning via Human Feedback (RLHF)☆4,755Jan 8, 2024Updated 2 years ago
- A LLaMA1/LLaMA12 Megatron implement.☆28Dec 13, 2023Updated 2 years ago
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆289Nov 24, 2025Updated 11 months ago
- The RedPajama-Data repository contains code for preparing large datasets for training large language models.☆4,998Jun 3, 2026Updated 5 months ago
- The TinyLlama project is an open endeavor to pretrain a 1.1B Llama model on 3 trillion tokens.☆9,010May 3, 2024Updated 2 years ago
- Official repository for LongChat and LongEval☆537May 24, 2024Updated 2 years ago
- train llama on a single A100 80G node using 🤗 transformers and 🚀 Deepspeed Pipeline Parallelism☆224Nov 21, 2023Updated 2 years ago
- AllenAI's post-training codebase☆3,915Updated this week
- Scalable toolkit for efficient model alignment☆850Oct 6, 2025Updated last year
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- YaRN: Efficient Context Window Extension of Large Language Models☆1,808Apr 17, 2024Updated 2 years ago
- MII makes low-latency and high-throughput inference possible, powered by DeepSpeed.☆2,116Jun 30, 2025Updated last year
- Large-scale, Informative, and Diverse Multi-round Chat Data (and Models)☆2,912Mar 13, 2024Updated 2 years ago
- LLM training code for Databricks foundation models☆4,462Mar 25, 2026Updated 7 months ago
- LongLLaMA is a large language model capable of handling long contexts. It is based on OpenLLaMA and fine-tuned with the Focused Transform…☆1,462Nov 7, 2023Updated 2 years ago
- Robust recipes to align language models with human and AI preferences☆5,707Sep 23, 2026Updated last month
- Train transformer language models with reinforcement learning.☆19,649Updated this week
- ⛷️ LLaMA-MoE: Building Mixture-of-Experts from LLaMA with Continual Pre-training (EMNLP 2024)☆1,004Dec 6, 2024Updated last year
- Instruction Tuning with GPT-4☆4,332Jun 11, 2023Updated 3 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Simple and efficient pytorch-native transformer text generation in <1000 LOC of python.☆6,267Aug 22, 2025Updated last year
- Reference implementation for DPO (Direct Preference Optimization)☆2,922Aug 11, 2024Updated 2 years ago
- Reaching LLaMA2 Performance with 0.1M Dollars☆989Jul 23, 2024Updated 2 years ago
- Benchmarking large language models' complex reasoning ability with chain-of-thought prompting☆2,773Aug 4, 2024Updated 2 years ago
- USP: Unified (a.k.a. Hybrid, 2D) Sequence Parallel Attention for Long Context Transformers Model Training and Inference☆696Sep 16, 2026Updated last month
- Implementation of paper Data Engineering for Scaling Language Models to 128K Context☆499Mar 19, 2024Updated 2 years ago
- An Easy-to-use, Scalable and High-performance Agentic RL Framework based on Ray (PPO & DAPO & REINFORCE++ & VLM & TIS & vLLM & Ray & Asy…☆10,136Oct 5, 2026Updated 3 weeks ago