China gains on the US in the artificial intelligence arms race