my-nixos-config/nixos/modules/gpu.nix

{ ... }: {
  hardware.graphics = {
    enable = true;
    enable32Bit = true;
  };
  nixpkgs.config.rocmSupport = true;
}