Use new flake generation

flake.nix

@@ -1,29 +1,14 @@
 {
-  description = "Flake for quantization kernels";
+  description = "Flake for activation kernels";
 
   inputs = {
-    flake-utils.url = "github:numtide/flake-utils";
-    nixpkgs.follows = "kernel-builder/nixpkgs";
-    kernel-builder.url = "path:/home/daniel/git/kernel-builder";
+    kernel-builder.url = "git+ssh://git@github.com/huggingface/kernel-builder";
   };
 
   outputs =
     {
       self,
-      flake-utils,
-      nixpkgs,
       kernel-builder,
     }:
-    flake-utils.lib.eachSystem [ flake-utils.lib.system.x86_64-linux ] (system: {
-
-      devShells = rec {
-        default = shells.torch24-cxx98-cu124-x86_64-linux;
-        shells = kernel-builder.lib.${system}.torchExtensionShells ./.;
-      };
-
-      packages = {
-        bundle = kernel-builder.lib.${system}.buildTorchExtensionBundle ./.;
-        redistributable = kernel-builder.lib.${system}.buildDistTorchExtensions ./.;
-      };
-    });
+    kernel-builder.lib.genFlakeOutputs ./.;
 }