GitHub · 灵魂体仓库·· 7 天前AI 评分40
mlops:面向 forward-only PyTorch 模型的独立操作与优化器库
als244/mlops
来源与原文
GitHub · 灵魂体仓库
发布时间:(北京时间)
阅读原始来源 · github.com ↗完整链接与内容说明
https://github.com/als244/mlops
本站目前仅有 AI 导读与摘要,未展示原文正文;技术细节、代码和完整结论请查看原始来源。
AI 导读
mlops 是一个独立库,为 forward-only PyTorch 模型提供三层张量操作接口:语义函数(如 mlops.rms_norm)、mlops.explicit 无状态 forward/backward 入口,以及 mlops.dispatch 精确实现选择,并附带 mlops.optim 下的无内存分配混合精度 AdamW。优化器只更新本地张量,梯度通信和参数分片由调用方负责。安装脚本支持 FlashAttention-3、flash-linear-attention、Quack、SonicMoE、tilelang 等实现,其中 FlashAttention-3 在 Hopper GPU 上自动启用,专家并行模块提供 Quack/TE MoE 和 LoRA 层。
来源:GitHub · 灵魂体仓库 · github.com