China is rapidly closing the US’s lead in AI research