找回密码
 立即注册

QQ登录

只需一步,快速开始

查看: 3|回复: 0

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

[复制链接]

6

主题

1

回帖

20

积分

新手上路

积分
20
发表于 昨天 20:51 | 显示全部楼层 |阅读模式
教学项目: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#


本帖子中包含更多资源

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

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

本版积分规则

Archiver|手机版|小黑屋|一起港湾 ( 青ICP备2025004122号-1 )

GMT+8, 2026-2-27 07:16 , Processed in 0.421555 second(s), 20 queries .

Powered by Discuz! X3.5

© 2001-2026 Discuz! Team.

快速回复 返回顶部 返回列表