TensorFlow.js

TensorFlow.js is an open-source hardware-accelerated JavaScript library for training and deploying machine learning models.

Develop ML in the Browser
Use flexible and intuitive APIs to build models from scratch using the low-level JavaScript linear algebra library or the high-level layers API.

Develop ML in Node.js
Execute native TensorFlow with the same TensorFlow.js API under the Node.js runtime.

Run Existing models
Use TensorFlow.js model converters to run pre-existing TensorFlow models right in the browser.

Retrain Existing models
Retrain pre-existing ML models using sensor data connected to the browser or other client-side data.

About this repo

This repository contains the logic and scripts that combine several packages.

APIs:

TensorFlow.js Core, a flexible low-level API for neural networks and numerical computation.
TensorFlow.js Layers, a high-level API which implements functionality similar to Keras.
TensorFlow.js Data, a simple API to load and prepare data analogous to tf.data.
TensorFlow.js Converter, tools to import a TensorFlow SavedModel to TensorFlow.js
TensorFlow.js Vis, in-browser visualization for TensorFlow.js models
TensorFlow.js AutoML, Set of APIs to load and run models produced by AutoML Edge.

Backends/Platforms:

TensorFlow.js CPU Backend, pure-JS backend for Node.js and the browser.
TensorFlow.js WebGL Backend, WebGL backend for the browser.
TensorFlow.js WASM Backend, WebAssembly backend for the browser.
TensorFlow.js WebGPU, WebGPU backend for the browser.
TensorFlow.js Node, Node.js platform via TensorFlow C++ adapter.
TensorFlow.js React Native, React Native platform via expo-gl adapter.

If you care about bundle size, you can import those packages individually.

If you are looking for Node.js support, check out the TensorFlow.js Node directory.

Examples

Check out our examples repository and our tutorials.

Gallery

Be sure to check out the gallery of all projects related to TensorFlow.js.

Pre-trained models

Be sure to also check out our models repository where we host pre-trained models on NPM.

Benchmarks

Local benchmark tool. Use this webpage tool to collect the performance related metrics (speed, memory, etc) of TensorFlow.js models and kernels on your local device with CPU, WebGL or WASM backends. You can benchmark custom models by following this guide.
Multi-device benchmark tool. Use this tool to collect the same performance related metrics on a collection of remote devices.

Getting started

There are two main ways to get TensorFlow.js in your JavaScript project: via script tags or by installing it from NPM and using a build tool like Parcel, WebPack, or Rollup.

via Script Tag

Add the following code to an HTML file:

<html>
  <head>
    <!-- Load TensorFlow.js -->
    <script src="https://cdn.jsdelivr.net/npm/@tensorflow/tfjs/dist/tf.min.js"> </script>


    <!-- Place your code in the script tag below. You can also use an external .js file -->
    <script>
      // Notice there is no 'import' statement. 'tf' is available on the index-page
      // because of the script tag above.

      // Define a model for linear regression.
      const model = tf.sequential();
      model.add(tf.layers.dense({units: 1, inputShape: [1]}));

      // Prepare the model for training: Specify the loss and the optimizer.
      model.compile({loss: 'meanSquaredError', optimizer: 'sgd'});

      // Generate some synthetic data for training.
      const xs = tf.tensor2d([1, 2, 3, 4], [4, 1]);
      const ys = tf.tensor2d([1, 3, 5, 7], [4, 1]);

      // Train the model using the data.
      model.fit(xs, ys).then(() => {
        // Use the model to do inference on a data point the model hasn't seen before:
        // Open the browser devtools to see the output
        model.predict(tf.tensor2d([5], [1, 1])).print();
      });
    </script>
  </head>

  <body>
  </body>
</html>

Open up that HTML file in your browser, and the code should run!

via NPM

Add TensorFlow.js to your project using yarn or npm. Note: Because we use ES2017 syntax (such as import), this workflow assumes you are using a modern browser or a bundler/transpiler to convert your code to something older browsers understand. See our examples to see how we use Parcel to build our code. However, you are free to use any build tool that you prefer.

import * as tf from '@tensorflow/tfjs';

// Define a model for linear regression.
const model = tf.sequential();
model.add(tf.layers.dense({units: 1, inputShape: [1]}));

// Prepare the model for training: Specify the loss and the optimizer.
model.compile({loss: 'meanSquaredError', optimizer: 'sgd'});

// Generate some synthetic data for training.
const xs = tf.tensor2d([1, 2, 3, 4], [4, 1]);
const ys = tf.tensor2d([1, 3, 5, 7], [4, 1]);

// Train the model using the data.
model.fit(xs, ys).then(() => {
  // Use the model to do inference on a data point the model hasn't seen before:
  model.predict(tf.tensor2d([5], [1, 1])).print();
});

See our tutorials, examples and documentation for more details.

Importing pre-trained models

We support porting pre-trained models from:

Various ops supported in different backends

Please refer below :

TFJS Ops Matrix

Find out more

TensorFlow.js is a part of the TensorFlow ecosystem. For more info:

For help from the community, use the tfjs tag on the TensorFlow Forum.
TensorFlow.js Website
Tutorials
API reference
TensorFlow.js Blog

Thanks, BrowserStack, for providing testing support.

Project Name	Stars	Repos Using This	Packages Using This	Most Recent Commit	Total Releases	Latest Release	Open Issues	License	Language
Tfjs	17,422	660	508	6 hours ago	126	August 23, 2022	564	apache-2.0	TypeScript
A WebGL accelerated JavaScript library for training and deploying ML models.
Keras Js	4,879	31	6	a year ago	7	February 09, 2018	83	mit	JavaScript
Run Keras models in the browser, with GPU support using WebGL
Hedgehog Lab	2,323			3 months ago			33	apache-2.0	TypeScript
Run, compile and execute JavaScript for Scientific Computing and Data Visualization TOTALLY TOTALLY TOTALLY in your BROWSER! An open source scientific computing environment for JavaScript TOTALLY in your browser, matrix operations with GPU acceleration, TeX support, data visualization and symbolic computation.
Car Simulator	319			6 years ago				mit	JavaScript
Autonomous car simulator (based on JavaScript & WebGL) implemented by fuzzy control system, genetic algorithm and particle swarm optimization.
Natml Unity	184			a month ago				apache-2.0	C#
High performance, cross-platform machine learning for Unity Engine. Register at https://hub.natml.ai
Webml Polyfill	153			2 months ago			328	apache-2.0	Python
Deprecated, the Web Neural Network Polyfill project has been moved to https://github.com/webmachinelearning/webnn-polyfill
Gammacv	151	1	9	14 days ago	19	March 27, 2023	11	mit	JavaScript
GammaCV is a WebGL accelerated Computer Vision library for browser
Tfjs Tutorials	59			3 years ago			2
📃 TensorFlow.js 官方指南繁體中文版
Natdevice	21			2 months ago			2	apache-2.0	C#
High performance, cross-platform media device streaming for Unity Engine. Register at https://hub.natml.ai
Wizmap	21			3 days ago				mit	TypeScript
Explore and interpret large embeddings in your browser with interactive visualization! 📍

Tfjs