Add build.toml

build.toml

@@ -0,0 +1,4 @@
+[general]
+name = "triton_scaled_mm"
+
+[torch]

flake.nix

@@ -0,0 +1,14 @@
+{
+  description = "Flake for Triton scaled matmul kernels";
+
+  inputs = {
+    kernel-builder.url = "git+ssh://[email protected]/huggingface/kernel-builder";
+  };
+
+  outputs =
+    {
+      self,
+      kernel-builder,
+    }:
+    kernel-builder.lib.genFlakeOutputs ./.;
+}