20 packages
@tensorflow-models/body-pix
Pretrained BodyPix model in TensorFlow.js
@tensorflow-models/coco-ssd
Object detection model (coco-ssd) in TensorFlow.js
@tensorflow-models/mobilenet
Pretrained MobileNet in TensorFlow.js
@tensorflow-models/posenet
Pretrained PoseNet model in TensorFlow.js
@tensorflow-models/universal-sentence-encoder
Universal Sentence Encoder lite in TensorFlow.js
@tensorflow/tfjs
An open-source machine learning framework.
@tensorflow/tfjs-automl
This packages provides a set of APIs to load and run models produced by AutoML Edge.
@tensorflow/tfjs-backend-cpu
Vanilla JavaScript backend for TensorFlow.js
@tensorflow/tfjs-backend-wasm
This package adds a WebAssembly backend to TensorFlow.js. It currently supports the following models from our [models](https://github.com/tensorflow/tfjs-models) repo: - BlazeFace - BodyPix - CocoSSD - Face landmarks detection - HandPose - KNN classifier
@tensorflow/tfjs-backend-webgl
GPU accelerated WebGL backend for TensorFlow.js
@tensorflow/tfjs-backend-webgpu
This package adds a GPU accelerated [WebGPU](https://www.w3.org/TR/webgpu/) backend to TensorFlow.js. It currently supports the following models: - BlazeFace - BodyPix - Face landmarks detection - HandPose - MobileNet - PoseDetection - Universal sentence
@tensorflow/tfjs-converter
Tensorflow model converter for javascript
@tensorflow/tfjs-core
Hardware-accelerated JavaScript library for machine intelligence
@tensorflow/tfjs-data
TensorFlow Data API in JavaScript
@tensorflow/tfjs-layers
TensorFlow layers API in JavaScript
@tensorflow/tfjs-node
This repository provides native TensorFlow execution in backend JavaScript applications under the Node.js runtime, accelerated by the TensorFlow C binary under the hood. It provides the same API as [TensorFlow.js](https://js.tensorflow.org/api/latest/).
@tensorflow/tfjs-node-gpu
For information please refer [here](https://github.com/tensorflow/tfjs/blob/master/tfjs-node/README.md).
@tensorflow/tfjs-react-native
TensorFlow.js platform implementation for React Native
@tensorflow/tfjs-tsne
TensorFlow.js powered tSNE implementation
@tensorflow/tfjs-vis
Utilities for in browser visualization with TensorFlow.js