Installation

blox works with all JAX transformations.

Prerequisites

Since blox uses JAX, check out the JAX installation instructions for your specific hardware (CPU/GPU/TPU).

You will need Python 3.11 or later.

Install from PyPI

To install the latest released version:

pip install jax-blox

Install from Source

To install the latest development version:

pip install git+https://github.com/hamzamerzic/blox.git