The rare US-China partnership comes as Apple prepares to roll out its on-device generative AI service in China.