Why are you saying this as if the AI would have control over the reactor.
It’s unlikely they’d even be in the same building, or even the same campus. We have these crazy things called “wires” that let us transmit a lot of power over distances, so your small nuclear reactor can be remote, safe and secure and your AI lab can just be on your main campus.