by okish on 12/16/23, 2:22 AM with 1 comments
by okish on 12/16/23, 2:22 AM
Learn how to speed up PyTorch code with custom Metal shaders to take advantage of MPS support on Apple silicon. We will write from scratch a Python library that compiles a Metal shader using C++/Objective-C and lets you call the method from Python using pybind11.