layer_norm / README.md
danieldk's picture
danieldk HF Staff
Update tag
5d26454 verified
metadata
tags:
  - kernels

This CUDA extension implements fused dropout + residual + LayerNorm from the flash-attention repo.