Frank A. Krueger personal blog
PyTorch for WebGPU
TL;DR I’ve been working on a WebGPU optimized inference and autograd library called webgpu-torch with an API that matches PyTorch. The goal is to run neural networks in the browser at speeds comparable to a Linux workstation. Many kernels have been implemented and its design is easily extensible. It’s available on NPM…
- Points
- 313
- HN comments
- 74
- Submissions
- 1
- Cited by
- 0 commenters
AI & Machine LearningSoftware Engineering2023
On Hacker News
| Submitted | Title on HN | Points | Comments |
|---|---|---|---|
| May 19, 2023 | PyTorch for WebGPU by mighdoll | 313 | 74 |
