torch.js is a WebGPU-accelerated tensor library with a PyTorch-compatible API. These docs cover everything from basic tensor operations to building and training neural networks in the browser.
New to torch.js? Start with the Getting Started guide, or jump into a package below.