To Stay Ahead of China in AI, the U.S. Needs to Work with China