TensorFlow.js is a JavaScript library for building, training and serving machine learning models. When running in the browser, it utilizes WebGL acceleration. TensorFlow.js is a part of the TensorFlow ecosystem. You can import pre-trained TensorFlow SavedModels and Keras models, for execution and retraining.
For more information on the API, follow the links to their Core and Layers repositories below, or visit js.tensorflow.org.
This repository contains the logic and scripts to form a union package, @tensorflowjs/tfjs, from
- TensorFlow.js Core, a flexible low-level API, formerly known as deeplearn.js.
- TensorFlow.js Layers, a high-level API modeled after Keras.