China is gaining ground in AI. But the U.S. still has a major advantage