Add Apple MPS (Metal) backend: native Metal kernels, autograd CPU bridges#1349

Open

dredozubov wants to merge 10 commits into

k2-fsa:masterfrom

dredozubov:feature/mps-backend

Commits on Mar 14, 2026

Add Apple MPS (Metal) backend: native Metal kernels, autograd CPU bridges, 65-test suite
dredozubov
committed
Fix PR review issues: flake8 violations, MPS sync guards, comments, copyright
dredozubov
committed
Fix cpplint line-length violations in mps_utils.h
dredozubov
committed
Fix cpplint line-length in pytorch_context.cu preprocessor directive
dredozubov
committed
Add missing #include <functional> to utils_inl.h (cpplint)
dredozubov
committed
Suppress cpplint include-order for torch/extension.h (intentional ordering)
dredozubov
committed
Fix cpplint line-length in mutual_information.cu
dredozubov
committed
Fix segfault on exit: use __unsafe_unretained for cached Metal pipeline states
dredozubov
committed
Docs: mention Apple Silicon MPS support in README and installation guide
dredozubov
committed
Fix swoosh functions on MPS: use ATen ops instead of K2_EVAL raw pointers
dredozubov
committed