Installation

Installation#

Getting Startedd#

To get started with XLB, you can install it using pip. There are different installation options depending on your hardware and needs:

Basic Installation (CPU-only)#

pip install xlb

Installation with CUDA support (for NVIDIA GPUs)#

This installation is for the JAX backend with CUDA support:

pip install "xlb[cuda]"

Installation with TPU support#

This installation is for the JAX backend with TPU support:

pip install "xlb[tpu]"

Notes:#

  • For Mac users: Use the basic CPU installation command as JAX’s GPU support is not available on MacOS

  • The NVIDIA Warp backend is included in all installation options and supports CUDA automatically when available

  • The installation options for CUDA and TPU only affect the JAX backend

To install the latest development version from source:

pip install git+https://github.com/Autodesk/XLB.git

The changelog for the releases can be found here.

For examples to get you started please refer to the examples folder.

Contents