Installation

From PyPI

To install the latest release version of the Jaxion package, run the following command:

pip install jaxion

For GPU support, use the following command instead:

pip install jaxion[cuda12]

Note

For now, to build with GPU support, use the build from source method below. This will be simplified in future releases.

Build from Source

Check out the repository:

git clone git@github.com:JaxionProject/jaxion.git

Navigate to the project directory:

cd jaxion

Install the package using pip (CPU version):

pip install .

For GPU support, use the following command instead:

pip install .[cuda12]

Verify the installation by running the test suite:

pytest