kernels-community
/

torch_harmonics_attn

Model card Files Files and versions

medmekk HF Staff commited on Oct 27

Commit

cde9e6f

·

verified ·

1 Parent(s): 84ec9f0

Update README.md

Files changed (1) hide show

README.md +5 -0

README.md CHANGED Viewed

@@ -1,3 +1,8 @@
 ## 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

+---
+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