Installation
============

**blox** works with all JAX transformations.

Prerequisites
-------------

Since **blox** uses JAX, check out the `JAX installation instructions <https://jax.readthedocs.io/en/latest/installation.html>`_ for your specific hardware (CPU/GPU/TPU).

You will need Python 3.11 or later.

Install from PyPI
-----------------

To install the latest released version:

.. code-block:: bash

   pip install jax-blox

Install from Source
-------------------

To install the latest development version:

.. code-block:: bash

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