TensorFlow backend for TensorFlow.js via Node.js
This repo is under active development and is not production-ready. We are
actively developing as an open source project.
Installing
npm install @tensorflow/tfjs-node
(or)
yarn add @tensorflow/tfjs-node
Before executing any TensorFlow.js code, load and set the backend to 'tensorflow'.
import * as tf from '@tensorflow/tfjs';
import '@tensorflow/tfjs-node';
tf.setBackend('tensorflow');
Development
yarn
yarn test
See the demo
directory that trains MNIST using TensorFlow.js with the
TensorFlow C backend.
cd demo/
yarn
yarn mnist
The important line to note is at the top of mnist.ts
, which sets the backend to
TensorFlow.
Optional: Build libtensorflow From TensorFlow source
This requires installing bazel first.
bazel build //tensorflow/tools/lib_package:libtensorflow
Supported Platforms
- Mac OS
- Linux
- Windows coming soon