查看: 84|回复: 0

教学项目:mamba3-minimal ,Mamba-3 架构的一个极简、单文件纯 PyTorch 实现版本。

[复制链接]

8

主题

2

回帖

28

积分

新手上路

积分
28
发表于 2026-2-26 20:51:47 | 显示全部楼层 |阅读模式
教学项目:mamba3-minimal ,Mamba-3 架构的一个极简、单文件纯 PyTorch 实现版本。

地址:  github.com/VikramKarLex/mamba3-minimal

该项目完整实现了 Mamba-3 的核心特性(如梯形离散化、复数 SSM/RoPE、MIMO 等),并且不依赖任何自定义的 C++ 或 Triton 内核,从而实现了硬件无关性,能够流畅运行在 NVIDIA GPU (CUDA)、Apple
Silicon (MPS) 以及普通 CPU 上。

#HOW I AI#


本帖子中包含更多资源

您需要 登录 才可以下载或查看,没有账号?立即注册

×
您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

关注公众号

相关侵权、举报、投诉及建议等,请发 E-mail:2776601884@qq.com

Powered by Discuz! X5.0 © 2001-2026 Discuz! Team.|青ICP备2025004122号-1

在本版发帖
关注公众号
返回顶部