PyRank

Metal Shaders Python Packages

Python packages with the GitHub topic metal-shaders. Sorted by monthly downloads, with stars and forks.
c0rmac
metal-linalg

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.

17K 14 0
asher
mlx-kquant

Native K-quant support for MLX, with a quantization and fine-tuning toolchain for Apple Silicon

11K 10 4
c0rmac
metal-linalg-torch

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.

2K 14 0