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

Name		Name	Last commit message	Last commit date
Latest commit History 6 Commits
assets		assets
benchmarks		benchmarks
csrc		csrc
evals		evals
mamba_ssm		mamba_ssm
tests/ops		tests/ops
.gitignore		.gitignore
README.md		README.md
rebuild.sh		rebuild.sh
setup.py		setup.py