WWDC24: Train your machine learning and AI models on Apple GPUs | Apple

Learn how to train your models on Apple Silicon with Metal for PyTorch, JAX and TensorFlow. Take advantage of new attention operations and quantization support for improved transformer model performance on your devices.

Discuss this video on the Apple Developer Forums:

00:00 - Introduction
01:36 - Training frameworks on Apple silicon
04:16 - PyTorch improvements
11:26 - ExecuTorch
13:19 - JAX features

More Apple Developer resources: