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

|

history blame contribute delete

224 Bytes

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