Installation

Requirements

JAX-HDC is currently in alpha and is not yet published to PyPI.

git clone https://github.com/rlogger/jax-hdc.git
cd jax-hdc
pip install -e .

For development (testing, linting, type checking):

pip install -e ".[dev]"

For running examples (matplotlib, scikit-learn):

pip install -e ".[examples]"

For reproducible development environments:

nix develop        # Enter development shell
nix build          # Build the package
nix run .#basic-operations
nix run .#classification-simple

JAX automatically uses GPU when available. For CUDA:

pip install --upgrade "jax[cuda12]"

For TPU, install JAX with TPU support as per the JAX installation guide.