This repo is under active development and is not production-ready. We will be actively developing this in open source. Stay tuned for an official release.
Currently, we only support developing TensorFlow.js code directly inside of this repository, however we plan on publishing on NPM soon.
# Download and install JS depencies, including libtensorflow 1.8.
yarn
# Publish the NPM locally for usage with other packages.
yarn publish-local
See the demo
directory that trains MNIST using TensorFlow.js with the
TensorFlow C backend.
cd demo/
yarn
# Link the tfjs-node NPM we published above.
yarn link-local @tensorflow/tfjs-node
# Run the training script. See demo/package.json for this script.
yarn mnist
The important line to note is at the top of mnist.ts
, which sets the backend to
TensorFlow.
bindTensorFlowBackend();
This requires installing bazel first.
bazel build //tensorflow/tools/lib_package:libtensorflow
- Mac OS
- Linux
- Windows coming soon