WebGL-accelerated ML // linear algebra // automatic differentiation for JavaScript.
Clone or download
nkreeger Upgrade webgl-ext type-defs to the latest version. (#1337)
* Upgrade webgl-ext type-defs to the latest version.

* Use specific versions
Latest commit 3270ccd Oct 23, 2018
Failed to load latest commit information.
.vscode Optimize the WebGL samplers (#1227) Oct 2, 2018
integration_tests/benchmarks Fix the firebase push script to actually call getFullYear() (#1310) Oct 11, 2018
scripts Chores: Cleans up dead scripts; Standardizes test file names (#1290) Oct 5, 2018
src Upgrade webgl-ext type-defs to the latest version. (#1337) Oct 24, 2018
.gitignore Collect benchmarks from karma and send them to firebase. (#1197) Aug 2, 2018
.npmignore improve .npmignore Jul 11, 2018
.npmrc initial commit Aug 3, 2017
.travis.yml Add linting to benchmarks and fix linter errors. (#1282) Sep 21, 2018
CONTRIBUTING.md Implement floordiv for integer division & update CONTRIBUTING.md. (#960) Apr 17, 2018
DEVELOPMENT.md Update readme (#910) Mar 30, 2018
ISSUE_TEMPLATE.md Issues Template: Redirects users to union repository. (#931) Apr 10, 2018
LICENSE initial commit Aug 3, 2017
PULL_REQUEST_TEMPLATE.md Update pull request template (#1050) May 29, 2018
README.md Remove deeplearnjs paragraph from README.md (#1157) Jul 11, 2018
karma.conf.js Optimize mobilnet_v2 on CPU by 50-100x by optimizing MathBackendCPU (#… Aug 6, 2018
package.json Upgrade webgl-ext type-defs to the latest version. (#1337) Oct 24, 2018
rollup.config.js build sourcemaps for esm bundle (#1160) Jul 12, 2018
tsconfig.json Remove the doc decorator in favor of a jsdoc annotation. (#1158) Jul 13, 2018
tslint.json Add tf.addN(), allow bool weights and improve tslint (#1190) Jul 26, 2018
yarn.lock Upgrade webgl-ext type-defs to the latest version. (#1337) Oct 24, 2018


TensorFlow.js Core API

A part of the TensorFlow.js ecosystem, this repo hosts @tensorflow/tfjs-core, the TensorFlow.js Core API, which provides low-level, hardware-accelerated linear algebra operations and an eager API for automatic differentiation.

Check out js.tensorflow.org for more information about the library, tutorials and API docs.

To keep track of issues we use the tensorflow/tfjs Github repo.


You can install TensorFlow.js via yarn or npm. We recommend using the @tensorflow/tfjs npm package, which gives you both this Core API and the higher-level Layers API:

import * as tf from '@tensorflow/tfjs';
// You have the Core API: tf.matMul(), tf.softmax(), ...
// You also have Layers API: tf.model(), tf.layers.dense(), ...

On the other hand, if you care about the bundle size and you do not use the Layers API, you can import only the Core API:

import * as tfc from '@tensorflow/tfjs-core';
// You have the Core API: tfc.matMul(), tfc.softmax(), ...
// No Layers API.

For info about development, check out DEVELOPMENT.md.

For more information

Thanks BrowserStack for providing testing support.