Apple could roll out AI features for iPhones in China as early as May