tensorflow/tfjs

A WebGL accelerated JavaScript library for training and deploying ML models.

解決する課題

TensorFlow.js は、開発者がブラウザや Node.js で直接機械学習モデルをトレーニング・デプロイできるようにし、各 ML タスクごとに専用のバックエンドサーバーを用意する必要をなくします。センサー情報などのクライアントサイドデータをサーバーに送らずにモデルの再学習に利用できます。

仕組み

低レベルの線形代数ライブラリまたは Keras に似た高レベルの Layers API を使って、ゼロからモデルを構築するための API を提供します。WebGL、WebGPU、WASM、Node.js 用のネイティブ C++ アダプタなど、複数のハードウェアアクセラレーションバックエンドをサポートし、パフォーマンスを確保します。さらに、既存の TensorFlow や Keras のモデルを JavaScript 環境で使用できるように変換・インポートするツールも含まれています。

対象ユーザー

Web アプリ、Node.js のサーバーサイドアプリ、または React Native のモバイルアプリに機械学習を組み込みたい JavaScript 開発者向けです。

ハイライト

  • マルチプラットフォーム対応：ブラウザ（WebGL、WebGPU、WASM 経由）と Node.js の両方で動作。
  • 柔軟な API：低レベルの数値計算と高レベルのモデル構築の両方を提供。
  • モデル変換：事前学習済みの TensorFlow SavedModel と Keras モデルをインポートして実行可能。
  • クライアントサイド学習：ローカルのセンサーやクライアントサイドデータを使った再学習をサポート。
  • 可視化：ブラウザ内でモデルを可視化する専用ライブラリを同梱。