Shumai

什么是Shumai？

Shumai 是一个开源的快速网络连接的可微张量库，适用于 TypeScript（和 JavaScript）。它基于 bun 和 Flashlight 构建，旨在为软件工程师和研究人员提供便捷。Shumai 旨在简化数据集的创建，加速小型模型的训练，使高级/细粒度训练/推理逻辑更加简洁，以及让构建应用变得愉快。

如何使用 Shumai？

安装 Bun 和 ArrayFire，然后使用 ‘bun install @shumai/shumai’ 安装 Shumai 包。使用标准数组工具并进行 JavaScript 原生数组之间的转换。利用梯度来训练模型。

Shumai 的核心功能

快速的张量操作
可微张量用于基于梯度的优化
网络连接功能
与 Bun 和 Flashlight 的集成
支持 CUDA（Linux）的 GPU 计算及 macOS 的 CPU 计算

Shumai 的使用案例

#1 创建和操作数据集
#2 训练小型机器学习模型
#3 实现高级的训练和推理逻辑
#4 构建需要张量计算的应用