Install on ROCm without CUDA wheels, and run the sLSTM kernel on ROCm - #126

Open

kashif wants to merge 4 commits into

kashif:rocm-support

Commits on Jul 20, 2026

don't pull cuda torch and nvidia wheels when installing on rocm or cpu
kashif
committed
2d1598e
build and run the slstm cuda kernel on rocm via hip
kashif
committed
a442e6e

Commits on Jul 21, 2026

install torch through backend extras instead of a forced cuda index
kashif
committed
2affde5
serialize the hipify cache across processes
kashif
committed
ffc4839