kernels-community
/

torch_harmonics_attn

Model card Files Files and versions

torch_harmonics_attn / README.md

medmekk's picture

medmekk HF Staff

Update README.md

cde9e6f verified about 2 months ago

|

224 Bytes

metadata

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