What's it going to take for the US to catch China in the AI race?