Apple Inc. is working to bring its AI features to China by the middle of this year, accelerating a complex undertaking that has required software changes and deep reliance on local partners.