disco icon indicating copy to clipboard operation
disco copied to clipboard
verified publisher icon epfml

look if we can benefit from WebGPU support

Open martinjaggi opened this issue 2 years ago • 0 comments

WebGPU just got released, and it might allow us to benefit from GPUs in the browser finally. TFJS actually also just started supporting it, at least with this new backend: https://www.npmjs.com/package/@tensorflow/tfjs-backend-webgpu

anyone interested to give this a spin, and see if it results in any ML training time speedups?

martinjaggi avatar Apr 07 '23 11:04 martinjaggi