1 files changed, 27 insertions, 0 deletions
diff --git a/flake.nix b/flake.nix
new file mode 100644
index 0000000..915e9f5
--- /dev/null
+++ b/flake.nix
@@ -0,0 +1,27 @@
+{
+  inputs = {
+    nixpkgs.url = "github:nixos/nixpkgs/nixos-unstable";
+  };
+
+  outputs = { self, nixpkgs, ... }:
+    let
+      pkgs = import nixpkgs { system = "x86_64-linux"; };
+    in {
+      devShells.x86_64-linux.default = pkgs.mkShell {
+        buildInputs = with pkgs; [
+          (python312.withPackages (python-pkgs: [
+            python-pkgs.numpy
+            python-pkgs.matplotlib
+            python-pkgs.graphviz
+          ]))
+
+          graphviz
+        ];
+
+        shellHook = ''
+          tmux -L autograd
+          exit
+        '';
+      };
+    };
+}