Installation

Prerequisites

Python 3.7+
PyTorch 1.9+
CUDA toolkit (for GPU acceleration)
Set CUDA_HOME environment variable before installing the kernels

Installation Options

Installation without custom CUDA kernels

git clone https://github.com/Belilovsky-Lab/pylo
cd pylo
pip install .

Installation with custom CUDA kernels

git clone https://github.com/Belilovsky-Lab/pylo
cd pylo
pip install .
python setup.py install --cuda

Install patch for MuP

python -m pylo.util.patch_mup