MegaDLMs

用扩散模型训练语言模型,更快更灵活

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 开源模型 🍴 复刻 35 🐛 待处理 3 👀 关注 4
访问官网 ↗ GitHub ↗
GitHub 星标 ★ 343
维护状态 低维护
是否开源 开源
定价模式 🏷️完全免费 开源框架,免费使用,需自行部署。

项目预览

MegaDLMs 项目界面预览图
图片来源:GitHub 官方仓库名片

项目数据

分类开源模型
开发团队JinjieNi
定价模式🏷️完全免费
价格说明开源框架,免费使用,需自行部署。
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
主要语言Python
技术栈/模型diffusion,large-language-models
GitHub 星标★ 343
上线时间2025-10-26 00:00:00
代码更新2025-11-11
仓库体积11.4 MB
本站数据更新2026-11-09
维护状态低维护
收录时间2026-08-09
浏览次数3

项目介绍

MegaDLMs 是一个面向扩散语言模型(DLM)训练的 GPU 优化框架,支持任意规模的模型训练。它解决了传统自回归语言模型训练效率低、难以并行化的问题,通过扩散过程实现非自回归生成,提升训练吞吐和推理灵活性。该框架是 Quokka、Super Data Learners 和 OpenMoE 2 等项目的训练后端,核心能力包括高效的数据加载、分布式训练策略、模型并行和内存优化,旨在让研究者和开发者能够轻松训练大规模扩散语言模型。项目基于 Python,技术栈围绕 PyTorch 生态,当前处于早期阶段,但已展现出在生成质量与训练速度上的潜力。

Star 增长趋势

本站每日记录该项目 GitHub 星标变化,观测越久曲线越完整

数据来源:GitHub API 逐日快照 · 收录于 2026-08-09 · 当前星标 ★ 343

核心亮点

  • 专为扩散语言模型设计,训练效率高于传统自回归方法
  • 支持从单卡到多节点任意规模扩展,适配不同算力环境
  • 已被多个知名项目(如 OpenMoE 2)采用,实战验证

不足之处

  • 早期项目,文档和社区生态尚不完善
  • 依赖特定硬件优化,普通 GPU 环境可能无法发挥全部性能

适用场景

  • 训练大规模扩散语言模型用于文本生成
  • 研究非自回归生成范式,提升推理速度
  • 作为基础框架支撑多模态或混合专家模型训练

替代项目

Hugging Face Transformers、DeepSpeed、FairSeq

上一篇:KnowledgeCircuits

下一篇:Time-LLM

同类项目推荐

katib 开源

在K8s上自动调参,让模型优化省心省力

WARNING: Katib project is DEPRECATED, users should migrate to OptimizationJob and Ku···

★ 1707 2026-08-09
NeoHorse 开源

让模型自己出题、自己训练、自己变强

NeoHorse-1: Towards Recursive Self-Improvement via Agentic Post-Training with Routin···

★ 1697 2026-09-22
ml-engineering 开源

大模型训练推理踩坑指南,照着调优省半年时间

Machine Learning Engineering Open Book

★ 19688 2026-08-09
Composition-RL 开源

把可验证提示词自动组合,低成本扩充 RL 训练数据

[Neurips 2026] Official repository for the paper "Composition-RL: Compose Your Verif···

★ 138 2026-10-05
SparseLLM 开源

全局剪枝大模型,高稀疏度下保住性能

Official Repo for SparseLLM: Global Pruning of LLMs (NeurIPS 2024)

★ 71 2026-08-09
KG-FIT 开源

用大模型给知识图谱做微调,让关系预测更准

[NeurIPS'24] Knowledge Graph Fine-Tuning using LLMs

★ 130 2026-08-09
SMILE-factory 开源

用普通显卡微调大模型,LoRA 省显存快速上手

Finetune Falcon, LLaMA, MPT, and RedPajama on consumer hardware using PEFT LoRA

★ 106 2026-10-01
llemr 开源

把病历变成指令,微调大模型看懂医疗数据

NeurIPS'24 DB (Spotlight) | Instruction Tuning Large Language Models to Understand E···

★ 62 2026-08-09