Made it work

modules/hardware/nvidia.nix

@@ -0,0 +1,17 @@
+{ config, ... }:
+
+{
+	hardware.graphics = {
+		enable = true;
+	};
+	services.xserver.videoDrivers = ["nvidia"];
+	nixpkgs.config.cudaSupport = true;
+	hardware.enableAllFirmware = true;
+	hardware.nvidia = {
+		modesetting.enable = true;
+		powerManagement.enable = false;
+		open = false;
+		nvidiaSettings = true;
+		package = config.boot.kernelPackages.nvidiaPackages.stable;
+	};
+}