Installation

Install and set up Python. There are various ways to do it; my favorite one is to use uv. Then:

pip install bartz

(Or the equivalent command for your Python installation manager, e.g., uv add bartz.) To install the latest development version, do instead

pip install git+https://github.com/bartz-org/bartz.git

To install a specific commit, do

pip install git+https://github.com/bartz-org/bartz.git@<commit hash>

bartz is meant to be used with GPUs. This requires to install special driver modules. To install support for a NVIDIA GPU, use one of these two extras:

pip install bartz[cuda12]
pip install bartz[cuda13]

The CUDA version can be read from the output of the command, e.g., nvidia-smi. For other GPUs, follow the installation instructions of jax.