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).