The Chinese technology is removed from most US government agencies as race for AI dominance steps up.