flake.nix


1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
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
        '';
      };
    };
}