/longhorn_cuda

CUDA kernels for the Longhorn Architecture

Primary LanguageCuda

Longhorn: State Space Models Are Amortized Online Learners

This repo contains the CUDA kernels for the Longhorn model.


Installtion

Please run the following commands to install the relevant dependencies:

git clone https://github.com/Cranial-XIX/longhorn_cuda.git
cd longhorn_cuda
chmod +x rebuild.sh && bash rebuild.sh