Kernels in kvcomp_cuda_extension folder

Setup

conda create -n kvcomp python=3.12
conda install nvidia/label/cuda-12.4.1::cuda-toolkit
pip install -r requirements.txt
cd kvcomp_cuda_extension
pip install .