torch-harmonics-kernels

Prebuilt CUDA kernels for torch-harmonics DISCO convolution and spherical neighborhood attention, packaged for the Hugging Face kernels ecosystem.

Included Functions

• disco_forward(...)
• disco_backward(...)
• attention_forward(...)
• attention_backward(...)
• preprocess_psi(...)

Project Notes

• This repository vendors the CUDA sources it needs and does not depend on the torch-harmonics source tree at build time.
• The current package is a first pass that wraps the existing CUDA kernels in a kernel-builder project.
• Python-side module integration and any higher-level autograd or fallback logic can live in a separate consumer package.