China is only 3-6 months behind US in AI, Trump official says