Shumai
Active标语：基于 Bun + Flashlight 的 TypeScript/JavaScript 快速可微分张量库
更新：-
社交媒体：
什么是 Shumai?
Shumai 是一款面向软件工程师和研究员的快速、支持网络连接的可微分张量库，适用于 TypeScript 和 JavaScript，基于 Bun 和 Flashlight 构建。可创建张量、执行矩阵运算、通过自动微分计算梯度，并在网络上部署模型。Linux 默认使用 CUDA GPU 计算；macOS 通过 ArrayFire 使用 CPU。MIT 开源，来自 Meta AI Research。
如何使用 Shumai?
安装 Bun 和 ArrayFire（macOS 用 brew install arrayfire，Ubuntu 用 apt 包）。运行 bun install @shumai/shumai。使用 import * as sm from '@shumai/shumai' 导入，用 sm.randn() 或 sm.tensor() 创建张量，调用 .backward() 计算梯度。联网张量示例见 facebookresearch.github.io/shumai 文档。
Shumai 核心特点
01
自动微分张量
为张量设置 requires_grad 并调用 backward() 自动填充梯度。支持 detach() 复制张量且不追踪梯度。
02
联网张量
用 sm.io.serve_model 部署模型，客户端通过 sm.io.remote_model 调用远程张量，实现分布式机器学习。
03
Bun + Flashlight 后端
基于 Bun 运行时和 Flashlight（ArrayFire）。Linux 默认 CUDA GPU；macOS 使用 ArrayFire CPU 后端。支持 matmul、randn、identity 及损失函数等标准运算。
04
JavaScript 原生数组
通过 toFloat32Array() 和 sm.tensor() 在 Float32Array 与张量间转换。标量张量可用 toFloat32() 转为 JavaScript 数字。
Shumai 使用案例
在 TypeScript 中快速原型化机器学习实验，无需离开 JS 生态
构建联网 ML 服务，支持远程张量调用和服务端自动微分
在 Bun 中进行可微分编程和基于梯度的优化研究
Shumai 常见问题解答
支持哪些平台？
如何安装 Shumai？
Shumai 是否可用于生产环境？
Shumai 产品背景及官方链接
Shumai 支持电子邮件和客户服务联系方式和退款联系方式等。
Shumai 公司信息
Shumai 公司名称： Meta Platforms, Inc.
Shumai 开发商类型： company
更多关于 Shumai, 请访问 关于我们页面(https://facebookresearch.github.io/shumai/)
Shumai GitHub
Shumai GitHub 链接： https://github.com/facebookresearch/shumai