Shumai

Shumaiとは何ですか？

Shumaiは、JavaScriptとTypeScript用の高速で微分可能なテンソルライブラリです。BunとFlashlightを利用して構築されており、ソフトウェアエンジニアや研究者にとって最適なツールです。まだ実験段階ですが、そのパフォーマンスと柔軟性は注目に値します。

Shumaiの特徴は何ですか？

• 高速なテンソル計算: GPUやアクセラレータを活用して高速な計算を実現。
• 微分可能なテンソル: 機械学習モデルのトレーニングに最適。
• JavaScript/TypeScript対応: 幅広い開発者層に対応。
• Bunとの連携: FFIバインディングが高速で、JSのオーバーヘッドを最小化。

Shumaiの使用例は何ですか？

• 機械学習モデルのトレーニング: 小規模モデルのトレーニングを高速化。
• データセットの作成: JavaScriptのネイティブ配列を活用してデータを効率的に処理。
• 複雑なトレーニングロジック: JSC JITコンパイラを利用して、高度なロジックを簡単に実装。

Shumaiの使い方は？

1. インストール: BunとArrayFireをインストールします。

   curl https://bun.sh/install | bash
   brew install arrayfire
   bun install @shumai/shumai
   

2. テンソルの作成と計算:

   import * as sm from "@shumai/shumai"
   let X = sm.randn([1024, 1024])
   let Y = X.matmul(sm.identity(1024))
   console.log(Y.shape)
   

3. 勾配計算:

   const W = sm.randn([128, 128])
   W.requires_grad = true
   const X = sm.randn([128, 128])
   const mse = X.sub(W).mul(X.sub(W)).sum()
   mse.backward()
   console.log(W.grad)