今天笨小兔分享一个快速的可微张量库,适用于 TypeScript 和 JavaScript。Shumai:属于AI开发者工具,AI研究工具,AI编程与Github等方面AI工具。
Shumai官网网址
点击访问:Shumai
Shumai: Shumai 是一个开源的快速网络连接的可微张量库,适用于 TypeScript(和 JavaScript)。它基于 bun 和 Flashlight 构建,旨在为软件工程师和研究人员提供便捷。Shumai 旨在简化数据集的创建,加速小型模型的训练,使高级/细粒度训练/推理逻辑更加简洁,以及让构建应用变得愉快。
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构建需要张量计算的应用
Shumai常见问题
下面是大家比较关心的一些问题解答。
什么是Shumai?
Shumai 是一个开源的快速网络连接的可微张量库,适用于 TypeScript(和 JavaScript)。它基于 bun 和 Flashlight 构建,旨在为软件工程师和研究人员提供便捷。Shumai 旨在简化数据集的创建,加速小型模型的训练,使高级/细粒度训练/推理逻辑更加简洁,以及让构建应用变得愉快。
如何使用 Shumai?
安装 Bun 和 ArrayFire,然后使用 'bun install @shumai/shumai' 安装 Shumai 包。使用标准数组工具并进行 JavaScript 原生数组之间的转换。利用梯度来训练模型。
Shumai 支持哪些平台?
Shumai 官方支持 macOS 和 Linux。通过 Docker + WSL2 + Linux 可以实现对 Windows 的支持。
Shumai 使用 GPU 还是 CPU 进行计算?
Linux 默认使用 CUDA 进行 GPU 计算,而 macOS 默认使用 CPU。未来计划支持 OpenCL。
我该如何安装 Shumai?
确保已安装 Bun,然后安装 ArrayFire。最后,使用 'bun install @shumai/shumai' 安装 Shumai 包。
什么是 Flashlight?
Flashlight 是一个快速、可扩展的机器学习引擎,用 C++ 编写,并用作 Shumai 的后端。