praeclarum/webgpu-torch

This is stars and forks stats for /praeclarum/webgpu-torch repository. As of 10 May, 2024 this repository has 475 stars and 11 forks.

webgpu-torch Tensor computation and autograd with WebGPU acceleration inspired by pytorch. Homepage: https://praeclarum.org/webgpu-torch Installation Webgpu-torch has no dependencies so you can just include it in your web page. <script src="https://cdn.jsdelivr.net/npm/webgpu-torch@latest/torch.js"></script> You can also include it using npm: npm i webgpu-torch Usage If you want to use WebGPU tensors, you must first initialize the library with: if (!await torch.initWebGPUAsync()) { ...