Skip to content
#

webgpu-compute

Here are 15 public repositories matching this topic...

PyTorch's API that trains in a browser tab on WebGPU — autograd, nn, optim, capture/replay, int8, ONNX out; held to real torch's values on 4,764 golden cases; measured on one page against TF.js, jax-js, Burn and ONNX Runtime Web (4–12× faster training, ahead of ORT at inference)

  • Updated Oct 3, 2026
  • Python

Add this topic to your repo

To associate your repository with the webgpu-compute topic, visit your repo's landing page and select "manage topics."

Learn more