Awesome. I may need to finally bite the bullet and upgrade my macOS to test out the MPP approach you've taken.

I've got a number of tiled-load kernels, and a top-k attention kernel that you might find interesting.