@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

npm i @tensorflow/tfjs-backend-webgpu

Metadata

Downloads