@tensorflow/tfjs-backend-webgpu

v4.21.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

Dependencies (1)

@tensorflow/tfjs-backend-cpu

Dev dependencies (12)

karma jasmine jasmine-core karma-jasmine karma-commonjs @babel/polyfill karma-typescript @tensorflow/tfjs-core karma-chrome-launcher karma-browserstack-launcher karma-jasmine-html-reporter karma-typescript-es6-transform
npm i @tensorflow/[email protected]

Metadata

Downloads