Kernels
kernel
topk / flake.nix
marcsun13's picture
marcsun13 HF Staff
topk-metal: top-k over a small row for MoE routers
5bb63f9
Raw History Blame Contribute Delete
307 Bytes
{
description = "Fused sequence-mixing kernels from ggml: the gated delta rule, as one kernel";
inputs = {
kernel-builder.url = "github:huggingface/kernels";
};
outputs =
{ self, kernel-builder }:
kernel-builder.lib.genKernelFlakeOutputs {
inherit self;
path = ./.;
};
}