QR, symmetric eigendecomposition and SVD for batches of matrices on Apple Silicon GPUs, for MLX and PyTorch. 2x to 10x gain in performance. Metal kernels per regime, routed to the GPU or the CPU by policies measured per Mac. C++, Python (pip: metal-linalg, metal-linalg-torch), Swift and C.
QR, symmetric eigendecomposition and SVD for batches of matrices on Apple Silicon GPUs, for MLX and PyTorch. 2x to 10x gain in performance. Metal kernels per regime, routed to the GPU or the CPU by policies measured per Mac. C++, Python (pip: metal-linalg, metal-linalg-torch), Swift and C.
