Sorry to say, but your "brain-hole" is wide open. "brain-hole wide open" = "脑洞大开", you know it is not offense but a joke.They will think out of the box for their nuclear propulsion systems. Whats to say they won't go the traditional nuclear propulsion route and just use the reactor along with existing AIP system. Basically a solid state heat/steam generating that an AIP system that is required to drive an electric engine !
More seriously, can you tell us how an AIP works? Which of the 3 types of AIP today can use a nuclear reactor as heat source without being classified as just another nuclear reactor?
Although everything is possible given enough time, but one has to walk on the ground.