The billion-dollar deal marks the latest effort from the Trump administration to get ahead in the AI arms race with China.