@tensorflow/tfjs-backend-webgpu

v4.22.0-rc.0
This package adds a GPU accelerated [WebGPU](https://www.w3.org/TR/webgpu/) backend to TensorFlow.js. It currently supports the following models: - BlazeFace - BodyPix - Face landmarks detection - HandPose - MobileNet - PoseDetection - Universal sentence

Readme Versions Dependencies Scripts

benchmark

yarn test --//:grep=benchmark

build-ci

yarn build

build

yarn --cwd .. bazel build tfjs-backend-webgpu:tfjs-backend-webgpu_pkg

bundle

yarn build

bundle-ci

yarn bundle

build-npm

yarn build

publish-npm

yarn --cwd .. bazel run tfjs-backend-webgpu:tfjs-backend-webgpu_pkg.publish

test

yarn --cwd .. bazel test tfjs-backend-webgpu:tfjs-backend-webgpu_test --test_output=streamed

test-dev

yarn --cwd .. bazel run tfjs-backend-webgpu:tfjs-backend-webgpu_test --test_output=streamed
npm i @tensorflow/tfjs-backend-webgpu

Metadata

Downloads