Tag
QuixiAI ported ThunderKittens to Metal, enabling kernel support on MPS and MLX for training models on Mac.
The author demonstrates that migrating a LivePortrait implementation from MPS to Apple's MLX framework on an M5 Max chip results in significantly better performance and speed.