US to tell dozens of partner countries to pick sides in AI race with China