MSLK

为 GenAI 训练推理准备的 PyTorch 高性能 GPU 算子集

开源 🏷️完全免费 🔧 仅开源项目,无在线服务,需自行部署 基础设施 🍴 复刻 86 🐛 待处理 35 👀 关注 4
GitHub ↗
GitHub 星标 ★ 159
维护状态 活跃
是否开源 开源
定价模式 🏷️完全免费 开源免费,无定价信息

项目预览

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

项目数据

分类基础设施
开发团队meta-pytorch
定价模式🏷️完全免费
价格说明开源免费,无定价信息
访问状态🔧 仅开源项目,无在线服务,需自行部署
是否开源是
开源协议NOASSERTION
主要语言Python
GitHub 星标★ 159
上线时间2025-11-03 00:00:00
代码更新2026-10-30
仓库体积3.0 MB
本站数据更新2026-10-31
维护状态活跃
收录时间2026-10-31
浏览次数0

项目介绍

MSLK(Meta Superintelligence Labs Kernels)是面向生成式 AI 训练与推理场景的 PyTorch GPU 算子库集合,目标是把大模型流程中频繁出现、但通用框架未必最优的关键算子单独做成高性能实现。核心能力包括 FP8 行级量化(row-wise quantization)以及集合通信(collective communications)等,前者用于降低显存占用与带宽压力、提升低精度训练和推理效率,后者服务于多卡多机并行训练中的通信环节。项目以 Python 为主,算子底层通常对接 CUDA,可直接在 PyTorch 生态中调用,适合需要在大规模模型训练或推理中压榨 GPU 性能的团队参考和复用。当前星标约 159,属于早期项目,接口和覆盖范围可能仍在演进,采用前建议先评估与自身 PyTorch 版本的兼容性及算子覆盖是否满足需求。

Star 增长趋势

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

数据来源:GitHub API 逐日快照 · 收录于 2026-10-31 · 当前星标 ★ 159

核心亮点

  • 聚焦 FP8 行级量化与集合通信等大模型高频瓶颈算子,针对性强
  • 以 PyTorch 算子库形式提供,便于在现有训练推理代码中直接集成
  • 由 Meta 相关团队设计,算子实现思路对自研优化有参考价值

不足之处

  • 项目处于早期,星标约 159,算子覆盖与接口稳定性有待观察
  • 文档与社区生态尚不成熟,遇到问题可参考的资料有限

适用场景

  • 大模型低精度训练中引入 FP8 量化算子降低显存与带宽开销
  • 多卡分布式训练中优化集合通信环节的性能
  • 推理服务中对量化等关键算子做针对性加速

替代项目

TransformerEngine、FlashAttention、DeepSpeed

上一篇:ESP32s3-LLM-Cluster

下一篇:hyperspace-db

同类项目推荐

universal 开源

用可定制数字类型,在精度与性能间自由取舍

Large collection of number systems providing custom arithmetic for mixed-precision a···

★ 505 2026-09-18
tokentab 开源

一条命令,算清所有 AI 编程工具花了多少钱

A CLI that reads Claude Code, Codex, and Gemini CLI session logs and works out how m···

★ 269 2026-08-09
Grandet 开源

让 AI Agent 自动找到并接入最便宜的模型 API

Find the cheapest API for any model and connect it from your AI agent: Grandet Skill···

★ 1011 2026-10-25
cerul 开源

让 AI 记住视频内容,秒级检索关键画面

Open-source video processing core and CLI. Search and annotate local videos with you···

★ 166 2026-08-09
milvus 开源

海量向量秒级搜索,轻松构建 AI 应用

Milvus is a high-performance, cloud-native vector database built for scalable vector···

★ 46535 2026-08-09
llm-bridge 开源

一套格式打通多家大模型,自带日志与错误兜底

Interoperability between input formats of various LLMs, with observability, error ha···

★ 325 2026-10-29
get-beam 开源

让 GPU 任务像调用 API 一样,按需弹性伸缩,省心省钱。

Run GPU inference and training jobs on serverless infrastructure that scales with yo···

★ 103 2026-08-09
chitu 开源

让大模型推理更快更省,轻松部署高并发服务

High-performance inference framework for large language models, focusing on efficien···

★ 2995 2026-08-09