Подозреваю, что там речь шла не о тренеровке модели, а об инференсе
ну мне кажется то что модуль процессора который может быстро пробежаться посчитать по нейронам готовой модели, может так же ускоренно считать и в режиме обучения?
нашел у них Beta версию ML ComputeFramework Accelerate training and validation of neural networks across the CPU and one or more GPUs. Судя по классам похоже на Keras на Swift’е 🙂 https://developer.apple.com/documentation/mlcompute. Но без примеров не разобраться.