Back to Libraries
TensorFlow.js
18k
Machine learning library for JavaScript
License
Apache-2.0
Released
2018
Stars
18k
Use Cases
Train ML models in browser, run pre-trained models, transfer learning, real-time predictions
Code Example
import * as tf from '@tensorflow/tfjs';
const model = tf.sequential();
model.add(tf.layers.dense({ units: 1, inputShape: [1] }));
model.compile({ loss: 'meanSquaredError', optimizer: 'sgd' });