dropout_layer_norm CUDA extension from csrc/layer_norm of the
odysseyml/flash-attention fork
(tag odyssey-v2.8.3-fused-1), repackaged in HF kernels build-variant layout.dropout_add_ln_fwd, dropout_add_ln_bwd,
dropout_add_ln_parallel_residual_fwd, dropout_add_ln_parallel_residual_bwd.1from kernels import get_kernel
2
3dropout_layer_norm = get_kernel("odyssey-systems/dropout-layer-norm")