Write Metal/MPS kernels for PyTorch operators. Use when adding MPS device support to operators, implementing Metal shad…
Write Metal/MPS kernels for PyTorch operators. Use when adding MPS device support to operators, implementing Metal shaders, or porting CUDA kernels to Apple Silicon. Covers native_functions.yaml dispatch, host-side operators, and Metal kernel implementation.
pytorch
cli
free
Others in the same category, ranked by how often they are opened.