{ pkgs, inputs, ... }: # The HIP extension has to be compiled against a ROCm build of PyTorch, and the # only ROCm torch in this flake with a cached binary is the one from the pinned # nixpkgs-torch211 input (2.11 + ROCm 7.2, python 3.13). Current nixos-unstable # ships torch 2.13, whose ROCm build fails to configure in nixpkgs and is not # on cache.nixos.org. inputs.nixpkgs-torch211.legacyPackages.${pkgs.stdenv.hostPlatform.system}.callPackage ./package.nix { }