torch-rbln 0.1.8

Rebellions Extension for PyTorch

PyTorch RBLN

About

PyTorch RBLN (torch-rbln) is a PyTorch extension that allows natural use of Rebellions NPU compute within PyTorch. By implementing eager mode, which operates in a define-by-run fashion, it supports the full lifecycle of model development, deployment, and serving in the PyTorch ecosystem. It is also convenient for debugging and related workflows.

The same interface style as CPU and GPU applies — the rbln device, torch.rbln, and torch.compile — so developers and customers can target RBLN NPUs with familiar APIs. Operations on rbln tensors are integrated via PyTorch’s out-of-tree extension path; execution is coordinated with the RBLN compiler and runtime (rebel-compiler).

PyTorch RBLN is currently in beta and under active development. APIs may change between releases, backward compatibility is not guaranteed, and production use is not recommended yet. For the full notice, architecture, supported operators, and tutorials, see PyTorch RBLN — Overview in the RBLN SDK documentation.

Getting started

Prerequisites

• Python 3.10–3.13 — see Developer guide — Requirements.
• rebel-compiler — required; not installed with torch-rbln. Use an RBLN Portal account and the RBLN package index as described in How to install.

Install pre-built wheels

torch-rbln (public wheel; torch resolves to 2.9.1+cpu via the PyTorch CPU index):

pip3 install torch-rbln --extra-index-url https://download.pytorch.org/whl/cpu

For rebel-compiler and the rest of the setup, see Prerequisites above and How to install.

Build from source

1. Install uv.
2. Follow Developer guide (venv, rebel-compiler, editable build).

git clone https://github.com/rbln-sw/torch-rbln.git
cd torch-rbln
uv venv .venv && source .venv/bin/activate
./tools/dev-setup.sh pypi

rebel-compiler must be available in the same environment before the torch-rbln build finishes (see Prerequisites).

Documentation

RBLN SDK (hosted)

• Overview — install, design, tutorials
• Developer guide — source build, versions, internal PyPI
• Troubleshooting — librbln / torch_rbln.diagnose, core dumps, logging, dtype / CPU, memory (maintained in the RBLN SDK docs)

This repository

• Configuration — environment variables and runtime options
• Test Guide — local test runs
• Linting — code style and lint
• Third-party update — PyTorch pin, upstream files, rebel-compiler version bumps in pyproject.toml

Contributing

See docs/CONTRIBUTING.md.

License

Apache License 2.0 — see LICENSE and NOTICE.

Contact

• Community: discuss.rebellions.ai
• Email: [e-mail]