The UAE and Saudi Arabia are investing billions in US AI infrastructure. The deals could help the US in the AI race against China.