Jax-metal: Accelerated Jax training on Mac

This is super cool. There are many JAX ports for popular models.

But I see a lot of HN users talking about Mac training/inference... And my hype flatlines when I go look up the prices for Apple's RAM configs.