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
index

torch-node API Reference

Complete API documentation for torch-node.

torch-node

Functions

getGPUBackend - Get the configured GPU backend from environment.
loadGPUProvider - Load the GPU provider based on environment configuration.
getGPUBackendInfo - Get information about the current GPU backend configuration.

Types

GPUBackend - Supported GPU backends.

torch-node.serialization

Functions

save - Save tensors or a model to a file.
load - Load tensors from a file.
load_safetensors - Load tensors from safetensors format with metadata.

Types

SerializationFormat - Supported serialization formats.
SaveOptions - Options for save.
LoadOptions - Options for load.