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