Tri-RMSNorm. Efficient kernel for RMS normalization with fused operations, includes both forward and backward passes, compatibility with PyTorch.

github.com/dtunai/Tri-RMSNorm

Vaya's read on this project

Problem, audience, market, and the verdict — sign in to see it.

Updates

No recent activity.