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

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?