metal-linalg

介绍

QR, symmetric eigendecomposition and SVD for batches of matrices on Apple Silicon GPUs, for MLX and PyTorch. 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.

翻译准备中

取自仓库 README

README

正在加载完整 README。 以下为已保存的摘录。

QR decomposition, symmetric eigendecomposition (eigh) and singular value decomposition (SVD) for batches of matrices on Apple Silicon GPUs, for MLX, PyTorch and plain float buffers. A C++ library, installed with Homebrew or built from source inside your own project, with Python packages for mlx.core arrays and for torch tensors, a C API, and a Swift package (on [Float], and on mlx-swift's MLXArray).