kernels-community/megablocks

1{2  description = "Flake for megablocks_moe kernel";3 4  inputs = {5    kernel-builder.url = "github:huggingface/kernel-builder";6  };7 8  outputs =9    {10      self,11      kernel-builder,12    }:13    kernel-builder.lib.genFlakeOutputs {14      path = ./.;15      rev = self.shortRev or self.dirtyShortRev or self.lastModifiedDate;16 17      pythonCheckInputs = pkgs: with pkgs; [ 18        tqdm19        py-cpuinfo20        importlib-metadata21        torchmetrics22      ];23    };24}