Greed.js is a javascript library that allows executing PyTorch code right in the browser. It uses the WebGPU of the browser to execute PyTorch operations.
We have written the core PyTorch operations in WGSL, which speedup the calculations.
Also for devices/browsers which don't have GPU support. We have implemented fallback to CPU with NumPy polyfills implementing the PyTorch functions.
Show HN: Greed.js – Run PyTorch code in browser using WebGPU | Heykuki News