triton-layer-norm / flake.nix
danieldk's picture
danieldk HF staff
Add Triton-based layer norm from flash-attention
0f75957
{
description = "Flake for Triton layer norm kernels";
inputs = {
kernel-builder.url = "github:huggingface/kernel-builder";
};
outputs =
{
self,
kernel-builder,
}:
kernel-builder.lib.genFlakeOutputs ./.;
}