How to run PyTorch, TensorFlow, and JAX on your Mac

I'll spoil it for you: Apple Silicon is fast!

If you have a Mac and don't use its GPU, I'm about to change your life.

I recorded a video with step-by-step instructions to show you how to run PyTorch, TensorFlow, and JAX on metal.

I also wrote some code so you can compare every framework running on your CPU and your GPU. Here is a link to the repository and the instructions.

I tested this on my M3 Max,…