Installation

Requirements

  • Python 3.9+

  • JAX 0.4.20+

  • jaxlib 0.4.20+

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

Installing from Source

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

For development (testing, linting, type checking):

pip install -e ".[dev]"

For running examples (matplotlib, scikit-learn):

pip install -e ".[examples]"

GPU/TPU

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.