Skip to main content

borch

PyTorch's shape, in a browser tab. Three implementations of one arithmetic — a numpy core (import borch as torch), a TypeScript runtime on WebGPU (borch-ts), and a Python binding over that runtime for Pyodide (borch_webgpu) — held to real PyTorch's values, errors and printed form within the range a curriculum uses.

  • See it run: https://playidea-lab.github.io/borch/site/ — the playground trains on your GPU; ten lessons and ten tutorials run every code block in the page.
  • The long document — how the values are guaranteed, the supported range, what is deliberately absent and why, borch.ts's design, conformance — is docs/BOOK.md. This page is the door; that one is the house.
  • For agents: AGENTS.md is the one page a coding agent needs — which of the three to use, ten rules that save a rewrite, three smoke tests. llms.txt at the site root lists these documents in the order to read them. In Claude Code: /plugin marketplace add playidea-lab/borch then /plugin install borch@borch gives the session the borch skill — the same page as a skill, with recipes. Any MCP-capable editor can add https://gitmcp.io/playidea-lab/borch as a server — GitMCP serves this repository's llms.txt and README; nothing to register. Context7 indexes it as /playidea-lab/borch.

What it is not

Not PyTorch. CUDA, distributed training, mixed precision and torch.compile are never coming — they cannot exist in a browser, or learning them means leaving it. An absent feature beats a wrong answer, so what is missing is written down as missing: tests/torch_gap.py prints the current count per namespace, and every gap carries a reason.

Thirty seconds

In a browser, nothing installed. Open the playground and press Run. The landing page times its own first run; measured nightly on a real adapter (tests/browser/first_run.py).

Python, on your machine.

uv pip install pyborch
import borch as torch

x = torch.tensor([1.0, 2.0, 3.0], requires_grad=True)
(x * x).sum().backward()
print(x.grad)           # tensor([2., 4., 6.])

TypeScript, in a page.

npm install borch-ts

Or a page with no build step. One file, opened from disk — no npm, no bundler, no server. The published package is on the CDNs as an ES module.

<script type="module">
  import { init, Tensor } from "https://cdn.jsdelivr.net/npm/borch-ts@0.4/+esm";
  await init();
  const x = Tensor.from([1, 2, 3], [3]);
  console.log(await x.mul(x).sum().toArray());   // Float32Array [14]
</script>

Checked nightly against the published package (npm run cdn:py), training loop included — what breaks this is a refactor, and every other check here reads the tree rather than the tarball.

import { init, Tensor } from "borch-ts";

await init();                          // asks for a WebGPU adapter, refuses a software one
const x = Tensor.from([1, 2, 3], [3]);
console.log(await x.mul(x).sum().toArray());   // Float32Array [14]

Both examples are run as written by the checks (tests/test_document_examples.py, borch-ts/test/readme.ts), so if they stop working the build says so.

Where it runs

Your machine What runs Measured
A WebGPU adapter — any GPU Everything: tensors, autograd, nn, training, ONNX export, the hub EfficientNet-B0 1.5 ms/image (apple / metal-3)
No adapter, and the page is cross-origin isolated — this site, site/serve.py, the offline bundle on localhost The cpu device on a pool of workers: pretrained backbones from the hub, a linear head on their features, cosine neighbours. The workbench's frozen path. B0 2.6 ms/image on 8 workers (M4 Max)
No adapter, plain page — file://, http:// from another machine, a browser without shared memory The same cpu device on one thread B0 14 ms/image (M4 Max)
A software adapter only (SwiftShader) All of the WebGPU surface at CPU speed — the badge says so — and the cpu device beside it, which is the faster of the two for a backbone B0 25× slower than the cpu device

The cpu device is not a second Tensor backend: it runs a checkpoint's bytes, not your code. Training a model of your own, the small CNN, ONNX export need the adapter. From Python it is import borch_cpu; the workbench's first cell falls to it by itself. The numbers and the reasons are in the book under The cpu device.

How it is guaranteed

4764 golden cases compare all three implementations against the same answers frozen from real torch — values, shapes, gradients, exception types and messages, and repr. The core runs them natively and in Pyodide; the binding and borch.ts run them in a browser on a real GPU, and every number they print carries the adapter's name, because a software adapter answers every WebGPU call correctly and proves nothing about a GPU.

Where this library deliberately parts from torch — a handful of places, each with a measurement beside it — is listed on the landing page and pinned by borch-ts/test/parity.ts. Everything else that differs is a defect, and the ledgers (tests/torch_gap.py, borch-ts/test/run.py) hold zero gaps without a reason.

Or a notebook. %pip install pyborch then import borch_webgpu as torch in any Pyodide — the notebook page is JupyterLite with the wheel already on its shelf, training on the tab's GPU 6 s after opening.

The file leaves as ONNX. onnx.exportOnnx(model, sample) in TypeScript, torch.onnx.export(model, x, path) in Python from borch_webgpu (the numpy core does not export) — traced from one forward, written without a dependency, and checked by ONNX Runtime Web reproducing the forward (3.5e-8; see the book).

Where things are

borch/ the numpy core — the reference implementation
borch-ts/src/ the TypeScript runtime: hand-written WGSL, zero dependencies
borch_webgpu/ the Python binding over borch.ts, for Pyodide
borchvision.py · borch-ts/src/vision.ts torchvision's transforms, ops, datasets
tests/ the golden, the ledgers, and the checks that police these documents
site/ the playground, lessons, tutorials, and the API reference
docs/BOOK.md the long document
ROADMAP.md what conformance means here, and what will not be done

Sister libraries: bimm (a model catalogue, on npm as bimm-ts) and borch-hub (weights by manifest and hash).

Working on it

Every browser check is listed in .github/workflows/gpu.yml and run nightly by tests/browser/nightly.py; CLAUDE.md holds the three rules a session has to know. Native tests: uv run --with pytest --with numpy --with torch --with torchvision --with scipy pytest tests/ -q.

Licence

Apache-2.0. Third-party notices are in THIRD-PARTY.md.

Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distributions

No source distribution files available for this release.See tutorial on generating distribution archives.

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

pyborch-1.12.0-py3-none-any.whl (1.1 MB view details)

Uploaded Python 3

File details

Details for the file pyborch-1.12.0-py3-none-any.whl.

File metadata

  • Download URL: pyborch-1.12.0-py3-none-any.whl
  • Upload date:
  • Size: 1.1 MB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: uv/0.9.16 {"installer":{"name":"uv","version":"0.9.16","subcommand":["publish"]},"python":null,"implementation":{"name":null,"version":null},"distro":{"name":"macOS","version":null,"id":null,"libc":null},"system":{"name":null,"release":null},"cpu":null,"openssl_version":null,"setuptools_version":null,"rustc_version":null,"ci":null}

File hashes

Hashes for pyborch-1.12.0-py3-none-any.whl
Algorithm Hash digest
SHA256 365b9c64a2ac5ae59f93f422ea29b44cbbfc2f3462bd42c82e46bc4e40035445
MD5 ec7f5428d0f893866be1fafd035b424e
BLAKE2b-256 a6595990982d9cf5294cace8786f0244d16c30e5c97a4ccce5200a66ca5dd3b7

See more details on using hashes here.

Release history Release notifications | RSS feed

This release

1.12.0 This release

1 file

1.11.3

1 file

1.11.2

1 file

1.11.1

1 file

1.11.0

1 file

1.6.0

2 files

1.4.0

2 files

Anthropic, PBC Visionary sponsor Bloomberg Visionary sponsor Hudson River Trading Visionary sponsor Meta Visionary sponsor NVIDIA Visionary sponsor Microsoft Sustainability sponsor Depot Continuous Integration AWS Cloud computing and Security Sponsor Datadog Monitoring Fastly CDN Google Download Analytics Sentry Error logging StatusPage Status page