dtunai / Tri-RMSNormView on GitHub
Efficient kernel for RMS normalization with fused operations, includes both forward and backward passes, compatibility with PyTorch.
12Jun 5, 2024Updated last year

Alternatives and similar repositories for Tri-RMSNorm

Users that are interested in Tri-RMSNorm are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.

Sorting:

Are these results useful?