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
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 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.
This requires installing bazel first.
bazel build //tensorflow/tools/lib_package:libtensorflow
- Mac OS
- Windows coming soon