Shumai 是一个开源的快速网络连接的可微张量库,适用于 TypeScript(和 JavaScript)。它基于 bun 和 Flashlight 构建,旨在为软件工程师和研究人员提供便捷。Shumai 旨在简化数据集的创建,加速小型模型的训练,使高级/细粒度训练/推理逻辑更加简洁,以及让构建应用变得愉快。
适合人群
- 软件工程师
- 研究人员
- 数据科学家
- 机器学习工程师
产品优势
- 与 TF.js 相比,性能更快
- 易于与 JavaScript 和 TypeScript 生态系统集成
- 支持 GPU 加速(Linux)
- 开源且积极开发中
使用注意
- 实验性软件,可能存在问题
- 平台支持有限(macOS 和 Linux)
- CPU 计算未得到良好优化
- 安装过程仍在不断完善中




