Skip to content
#

amx

Here are 99 public repositories matching this topic...

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.

  • Updated Oct 10, 2026
  • C

Add this topic to your repo

To associate your repository with the amx topic, visit your repo's landing page and select "manage topics."

Learn more