pkgs: self: super: with self; {

  # Backport upstream master tensorflow (bazel build is failing)
  astunparse = callPackage ./tf-astunparse.nix { };

  gast = callPackage ./tf-gast.nix { };

  tensorflow-tensorboard = callPackage ./tf-tensorboard.nix {
    protobuf = protobuf3_5;
  };

  tensorflow = callPackage ./tf.nix {
    protobuf = protobuf3_5;
    cudaSupport = pkgs.config.cudaSupport or false;
    cudatoolkit = pkgs.cudatoolkit90;
    cudnn = pkgs.cudnn_cudatoolkit90;
    inherit (pkgs.linuxPackages) nvidia_x11;
  };

  # https://github.com/NixOS/nixpkgs/pull/26580 (Cuda definitions are backwards too)
  libgpuarray = callPackage ./theano-libgpuarray.nix {
    clblas = pkgs.clblas.override { inherit boost; };
    cudaSupport = pkgs.config.cudaSupport or false;
    inherit (pkgs.linuxPackages) nvidia_x11;
  };

  Theano = callPackage ./theano.nix rec {
    cudaSupport = pkgs.config.cudaSupport or false;
    cudnnSupport = cudaSupport;
    cudatoolkit = pkgs.cudatoolkit8;
    cudnn = pkgs.cudnn_cudatoolkit8;
    inherit (pkgs.linuxPackages) nvidia_x11;
  };

  TheanoWithoutCuda = super.TheanoWithoutCuda.override {
    cudaSupport = false;
    cudnnSupport = false;
  };

  TheanoWithCuda = super.TheanoWithCuda.override {
    cudaSupport = true;
    cudnnSupport = true;
  };
}