News for hackers. Stuff that matters.

A comment on Batmobile: 10-20x Faster CUDA Kernels for Equivariant Graph Neural Networks. Parent comment

Re:Batmobile: 10-20x Faster CUDA Kernels for Equivariant Gra... by teddykoker on Wednesday January 21, 2026 @08:57AM (#46705776)
OpenEquivariance [1] is another good baseline for with kernels for the Clebsch-Gordon tensor product and convolution, and it is fully open source. Both kernel implementations have been successfully implemented into existing machine learning interatomic potentials, e.g. [2,3].

[1] https://github.com/PASSIONLab/OpenEquivariance

[2] https://arxiv.org/abs/2504.16068

[3] https://arxiv.org/abs/2508.16067

[ Parent | on HN ]

There are no comments on this comment yet.