Settings

Theme

Metal shaders with PyTorch from end to end

medium.com

4 points by okish 2 years ago · 1 comment

Reader

okishOP 2 years ago

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.

Keyboard Shortcuts

j
Next item
k
Previous item
o / Enter
Open selected item
?
Show this help
Esc
Close modal / clear selection