Dawn
The @torchjsorg/dawn package provides an alternative WebGPU backend for Node.js environments based on Google's Dawn project (the same engine used in Chrome).
Why choose Dawn?
- Chrome Parity: If your model must behave exactly like it does in the browser, Dawn offers the highest compatibility.
- Robustness: Benefit from the massive engineering effort put into the Chrome browser's WebGPU implementation.
Installation
npm install @torchjsorg/dawn;Usage
import { dawn } from '@torchjsorg/dawn';
import torch from '@torchjsorg/torch.js';
// Register Dawn as the global WebGPU provider
torch.webgpu.setProvider(dawn);