---
tags:
- kernels
- cuda
---
## Torch Harmonics Attn

Attention mechanisms for the Spherical Harmonics basis using the torch-harmonics package : https://github.com/NVIDIA/torch-harmonics/tree/main/torch_harmonics/attention