tensorflow/tfjs

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

它解决了什么问题

TensorFlow.js 让开发者能够直接在浏览器或 Node.js 中训练和部署机器学习模型，免去为每个 ML 任务都需要专用后端服务器的需求。它支持使用客户端数据（如传感器数据）重新训练模型，而无需将数据发送到服务器。

它如何工作

它提供一套 API，允许使用低层线性代数库或类似 Keras 的高层 Layers 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 模型。
  • 客户端训练：支持使用本地传感器或客户端数据重新训练模型。
  • 可视化：内置专用的浏览器内模型可视化库。