As I understand, many people in Apple’s ML org come from Google. So there’s a bias to use Google software and hardware. Moreover, there’s probably a deal that Apple extracted to do training on Google hardware.

The important thing for Apple is the endpoint inference which will be done on Apple Silicon.