Installation

HeteroSymNN is designed to be lightweight and portable. By default, it runs in pure Python mode with no heavy dependencies.

Standard Installation (CPU)

For basic usage or if you have a C++ compiler installed (for CPU JIT acceleration):

pip install HeteroSymNN

Note

For CPU JIT acceleration, a C++ compiler (g++, clang, or cl.exe) is optional but highly recommended.

GPU Installation (CUDA)

To enable the high-performance CUDA backend using CuPy, install with the gpu extra:

pip install HeteroSymNN[gpu]

Note

For CuPy to work you need the Cuda drivers installed.

From Source

git clone https://github.com/Dilosch03/HeteroSymNN.git
cd HeteroSymNN
pip install -e .