| Python | 3.10 or newer (3.10, 3.11 and 3.12 are declared). |
| PyTorch | Installed by pip as a dependency. On a GPU machine, install the right build first — see Install by hardware. |
| Disk | The engine and its dependencies are large (about 5.5 GB measured on 0.5.2 with the default CUDA build of PyTorch), plus roughly twice the size of each model you import. See Troubleshooting. |
| OS | Linux (full). Windows with CUDA (Triton is not available on Windows). macOS on Apple Silicon through MPS. |
| Driver / compute capability | No minimum is stated yet. Run neurobrix doctor. |
Debian, Ubuntu, Fedora and other distributions that follow PEP 668 refuse a plain pip install (error: externally-managed-environment). Use a virtual environment:
python3 -m venv nbx
source nbx/bin/activate
pip install neurobrix
Expected result: Successfully installed ... neurobrix-0.6.2.
On Debian and Ubuntu, python3 -m venv may need one package first:
sudo apt update && sudo apt install -y python3-venv
pipx install neurobrix
neurobrix --version
Expected result: NeuroBrix v0.6.2.
neurobrix doctor
Expected result: No blocking problem found and exit code 0. If it lists a problem, it prints the command that fixes it. doctor checks PATH, the PyTorch build, Triton, NVIDIA GPUs and their driver. It does not check AMD or Apple devices. Use neurobrix doctor --json in scripts.
The default pip install neurobrix takes whatever PyTorch build the package index serves. That is not always the right one for your card. Pick your case in Install by hardware.
Triton is not a declared dependency of neurobrix. Without it, --triton is unavailable and the default compiled engine still works.
pip install "neurobrix[adapter]"
The adapter extra adds tokenizer and audio libraries. Which models need it is not documented yet. There is no cuda or cpu extra.
~/.neurobrix/
├── store/ # downloaded .nbx archives (and .part files while downloading)
└── cache/ # extracted models — this is what the runtime loads
Both live under the invoking user's home. A model imported as one user is not visible to another, including under sudo.
Install by hardware · then the first hour
© 2025–2026 Hocine Benkelaya. NeuroBrix engine released under the Apache License 2.0.
Developed by Hocine Benkelaya · Supported and funded by Neural Network Holding Ltd · Commercial distribution by WizWorks OÜ