Nuclear weapons come with the immense benefit that purifying uranium is so challenging, expensive, and difficult to hide that we don't even bother to protect the information on how to build a nuke. They also have a second benefit of being impossible to test without the whole world knowing you did.
AI has no such hurdles.
No, the AI compute supply chain is actually quite similar to the nuclear supply chain. Building chips capable of training or large-scale inference is extraordinarily complex, expensive, has a large footprint, and has a few chokepoints (rare earths mining, HBM manufacturing, and ultraviolet lithography, where there is literally one company in the world capable of building the relevant outputs) where we in fact already have governance mechanisms for things like export controls.