HEJAX

A GPU-accelerated homomorphic encryption library built with JAX. It supports the CKKS scheme for approximate number computations.

Installation

You can install the library locally using pip. It is recommended to do this in a virtual environment.

# Clone the repository (if you haven't already)
git clone <repository-url>
cd hejax

# Install dependencies and the library in editable mode
pip install -r requirements.txt
pip install -e .

Usage

Here is a basic example of how to use the library. For more detailed examples, please see the notebooks in the examples directory.

# (A simple, illustrative example will go here)
# TODO: Add a simple usage example.

Running the Examples

The examples directory contains a series of Jupyter notebooks that demonstrate the library's features.

After installing the requirements, you can run Jupyter Lab:

jupyter lab

Then, navigate to the examples directory to open and run the notebooks.