MegaDLMs

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

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

开源 free 开源模型
访问官网 ↗ GitHub ↗ 文档 ↗
GitHub 星标 ★ 344
维护状态 低维护
是否开源 是
定价模式 free

项目数据

分类开源模型
开发团队JinjieNi
所属国家
定价模式free
价格说明开源框架,免费使用,需自行部署。
访问状态
是否开源是
开源协议
主要语言Python
技术栈/模型diffusion,large-language-models
GitHub 星标★ 344
30天Star增速
HF 下载量
上线时间2025-10-26 00:00:00
最近更新2026-08-29 00:00:00
维护状态低维护
中文支持
访问方式
移动端支持
综合评分
收录时间2026-08-09
浏览次数2

使用教程

核心亮点

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

不足之处

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

适用场景

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

替代项目

Hugging Face Transformers、DeepSpeed、FairSeq

项目介绍

MegaDLMs 是模型训练领域的开源项目,由 JinjieNi 开发,2025 年首次发布。

它收录于模型训练分类,该分类目前共有 522 个项目,GitHub 星标数为 344。

项目已超过三个月没有代码更新,维护节奏明显放缓,最近一次代码更新于 2026-08-29。免费使用,需自行部署。

它主要面向的使用场景是:训练大规模扩散语言模型用于文本生成。同类可对比的替代方案包括 Hugging Face Transformers、DeepSpeed、FairSeq。

上一篇:KnowledgeCircuits

下一篇:Time-LLM

同类项目推荐

Long-RL 开源

让强化学习轻松驾驭超长序列,训练更稳更快

Long-RL: Scaling RL to Long Sequences (NeurIPS 2025)

★ 729 2026-08-09
TPA 开源

把注意力复杂度从平方降到线性,长序列不再卡顿

[NeurIPS 2025 Spotlight] TPA: Tensor ProducT ATTenTion Transformer (https://arxiv.or···

★ 463 2026-08-09
minimind 开源

两小时造出你的小模型,练手入门不求人

Train a 64M-parameter LLM from scratch in just 2h!

★ 62178 2026-08-09
pruna 开源

一键优化 AI 模型,推理提速又省资源。

Pruna is a model optimization framework built for developers, enabling you to delive···

★ 1286 2026-08-13