Optimum RBLN

🤗 Optimum RBLN provides an interface between HuggingFace libraries (Transformers, Diffusers) and RBLN NPUs, including ATOM and REBEL.

This library enables seamless integration between the HuggingFace ecosystem and RBLN NPUs through a comprehensive toolkit for model loading and inference across single and multi-NPU environments. While we maintain a list of officially validated models and tasks, users can easily adapt other models and tasks with minimal modifications.

Key Features

🚀 High Performance Inference

Optimized model execution on RBLN NPUs through RBLN SDK compilation
Support for both single and multi-NPU inference
Integrated with RBLN Runtime for optimal performance

🔧 Easy Integration

Seamless compatibility with HuggingFace Model Hub
Drop-in replacement for existing HuggingFace pipelines
Minimal code changes required for NPU acceleration

Seamless Replacement for Existing HuggingFace Code

Swap the HuggingFace class for its RBLN counterpart. Passing a HuggingFace model id compiles the model for the NPU on the first run; passing a directory of previously compiled artifacts loads them directly, so compilation is skipped.

- from diffusers import StableDiffusionXLPipeline
+ from optimum.rbln import RBLNStableDiffusionXLPipeline

# Load model
model_id = "stabilityai/stable-diffusion-xl-base-1.0"
prompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k"
- pipe = StableDiffusionXLPipeline.from_pretrained(model_id)
+ pipe = RBLNStableDiffusionXLPipeline.from_pretrained(model_id)

# Generate image
image = pipe(prompt).images[0]

# Save image result
image.save("image.png")

+ # (Optional) Save compiled artifacts to skip the compilation step in future runs
+ pipe.save_pretrained("compiled_sdxl")

Compile Ahead of Time with the CLI

Instead of compiling inside your script, you can compile a model up front with the optimum-rbln-cli command and load the resulting artifacts later:

# Compile and save the artifacts to ./compiled_qwen3
optimum-rbln-cli --model-id Qwen/Qwen3-4B -o ./compiled_qwen3 \
    --max_seq_len 8192 --batch_size 1 --num_devices 4

from optimum.rbln import RBLNQwen3ForCausalLM

# Load the compiled artifacts (no recompilation)
model = RBLNQwen3ForCausalLM.from_pretrained("./compiled_qwen3")

Useful CLI helpers:

optimum-rbln-cli --list-classes                     # list available RBLN classes
optimum-rbln-cli --class RBLNQwen3ForCausalLM --show-rbln-config  # show accepted rbln_config keys
optimum-rbln-cli --examples                          # show more usage examples

Documentation

Check out the documentation of Optimum RBLN for more advanced usage.

Getting Started

Note: The rebel-compiler library, which is required for running optimum-rbln, is only available for approved users. Please refer to the installation guide for instructions on accessing and installing rebel-compiler.

Install from PyPI

To install the latest release of this package:

pip install optimum-rbln --extra-index-url https://download.pytorch.org/whl/cpu

Install from source

Prerequisites

Install uv (refer to this link for detailed commands)

The below command installs optimum-rbln along with its dependencies.

git clone https://github.com/rbln-sw/optimum-rbln.git
cd optimum-rbln
./scripts/uv-sync.sh

Need Help?

Join discussions and get answers in our Developer Community
Contact maintainers at support@rebellions.ai

Metadata

Release files for optimum-rbln 0.12.1.dev0

For a detailed explanation of source distributions (sdists) and built distributions (wheels), please see the package formats documentation.

Source distribution (sdist)

Source distribution for optimum-rbln 0.12.1.dev0
File	Size	Uploaded
optimum_rbln-0.12.1.dev0.tar.gz	680.4 kB	Oct 2, 2026	Details

Built distribution (wheel)

Table of built distributions (wheels) for optimum-rbln 0.12.1.dev0
File	Interpreter	ABI	Platform	Reset
optimum_rbln-0.12.1.dev0-py3-none-any.whl 720.9 kB Oct 2, 2026	Python 3	none	any	Details

Total release size: 1.4 MB

Release files / optimum_rbln-0.12.1.dev0.tar.gz

Download URL	optimum_rbln-0.12.1.dev0.tar.gz
Size	680.4 kB
Tags	Source
SHA-256 checksum How to use checksums	`d205279c8f1850e438a10821fa99a2ed58d74f8953bccffb6d7326b4998e95d5`
BLAKE2b-256 checksum How to use checksums	`739a3001063e714e1857e3d47327a729b6c926e5fd550c452b4006d8e0275ba6`
Upload date	Oct 2, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/6.2.0 CPython/3.12.3`

Release files / optimum_rbln-0.12.1.dev0-py3-none-any.whl

Download URL	optimum_rbln-0.12.1.dev0-py3-none-any.whl
Size	720.9 kB
Tags	Python 3
SHA-256 checksum How to use checksums	`1fb0b5e83302ecf4a241c6e8babd123b207b38ec5a9e2544d19f59f8e0c563aa`
BLAKE2b-256 checksum How to use checksums	`b38caecc68528537768140484032f1778998d09ee64ee08acc270675ee070f21`
Upload date	Oct 2, 2026
Uploaded using Trusted Publishing? What is trusted publishing?	No
Uploaded via	`twine/6.2.0 CPython/3.12.3`