{
  description = "DAWG Flake";

  inputs = {
    pyproject-nix.url = "github:pyproject-nix/pyproject.nix";
    pyproject-nix.inputs.nixpkgs.follows = "nixpkgs";
  };
  outputs = {
    nixpkgs,
    pyproject-nix,
    ...
  }: let
    system = "x86_64-linux";
    project = pyproject-nix.lib.project.loadPyproject {
      projectRoot = ./.;
    };
    pkgs = import nixpkgs {
      inherit system;
      config = {
        allowUnfree = true;
        cudaSupport = true;
      };
    };

    python = pkgs.python3;
    cudaPackages = pkgs.cudaPackages;
  in {
    devShells.x86_64-linux.default = let
      arg = project.renderers.withPackages {inherit python;};
      pythonEnv = python.withPackages arg;
    in
      pkgs.mkShell {
        packages =
          [pythonEnv]
          ++ (with pkgs; [
            basedpyright
            black

            ffmpeg
            playerctl
            portaudio
            ripgrep

            gcc
            stdenv.cc.cc.lib

            cudaPackages.cudatoolkit
          ]);

        shellHook = ''
          export CUDA_HOME=${cudaPackages.cudatoolkit}
          export CUDA_PATH=${cudaPackages.cudatoolkit}
          export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/run/opengl-driver/lib:/run/opengl-driver-32/lib
        '';
      };

    packages.x86_64-linux.default = let
      attrs = project.renderers.buildPythonPackage {inherit python;};
    in
      python.pkgs.buildPythonPackage attrs;
  };
}