INFO:
Learn how to train your models on Apple Silicon with Metal for PyTorch, JAX and TensorFlow. Take advantage of new attention operations...
Train your machine learning and AI models on Apple GPUs - WWDC24 - Videos - Apple Developer