China now leads the U.S. in this key part of the AI race