Nuclear technology is tricky to get working even if you have decent scientists and all the published research about basic principles. For current ML tech, a big point is that the core techniques are really simple, the simple textbook methods work at large scale, you don't necessarily need any special sauce which could be restricted. Yes, there is a lot of engineering work that goes into a full system to scale it efficiently, but all of that can be replicated by any decent bunch of engineers, unlike nuclear weapons.
The cat is out of the bag; you can't put the genie back in the bottle, all the "sensitive knowledge" you'd need to restrict AI is already known by at least a few CS undergrads in every single university in the world.