flashinfer.norm# Kernels for normalization layers. rmsnorm(input, weight[, eps, out]) Root mean square normalization.