drmerlot/custom-pytorch-metal-ops-demo

This repo contains some demos, explanations, and tutorials showing how to create custom pytorch ops on apple silicon for the MPS device with metal shaders.

Jupyter NotebookMIT