US risks losing AI arms race to China and Russia