Skip to main content

torch.js has not been released yet.

Playground Contact

Introduction Type Safety Tensor Expressions Tensor Indexing Einsum Einops Autograd Training a Model Profiling & Memory PyTorch Migration Best Practices Runtimes Performance PyTorch Compatibility Benchmarks DType Coverage

Introduction Quickstart Client API Worker API File I/O Backends

getGPUBackend loadGPUProvider getGPUBackendInfo GPUBackend save load load_safetensors SerializationFormat SaveOptions LoadOptions

torch.js· 2026

Legal Terms of Use Privacy Policy

/

/

docs
torch-node
torch-node
GPUBackend

torch-node.GPUBackend

export type GPUBackend = 'wgpu-native' | 'dawn';

Values

'wgpu-native'
'dawn'

Supported GPU backends.

getGPUBackendInfo

loadGPUProvider