US at risk of losing AI arms race to China and Russia