mlsanityData to 7 Oct 2026

Lightweight Gaussian Process Inference in C++ on Metal and CUDA

arXiv 2605.17898 ↗First posted 18 May 2026cs.LGcited by 0

Not on the map. Why?

Abstract (arXiv)

Gaussian process (GP) inference in Python is dominated by libraries such as GPyTorch and GPflow, which are built on deep-learning frameworks and inherit their dispatch overhead and dependency footprint. We present LightGP, a dependency-free C++17 library for GP regression with Python bindings, supporting Apple Metal and NVIDIA CUDA backends alongside tuned CPU paths via Apple Accelerate and OpenBLAS. LightGP provides four inference paths -- exact Cholesky, matrix-free conjugate gradients, sparse variational free energy, and structured kernel interpolation with FFT -- covering problems from N=100 to N=500,000. On an Apple M4, LightGP CPU is 2.6--8.7× faster than GPyTorch CPU for exact GP and ~1.5× faster for sparse GP at every scale tested. On an NVIDIA RTX~3060, LightGP CUDA is 2.3--6.7× faster than GPyTorch CUDA for exact GP up to N=2,048, with GPyTorch closing the gap at N=4,096. A fused matrix-free kernel-vector product on Metal achieves 32× over the explicit path at N=20,000 with O(N) memory, and an FFT-accelerated SKI matvec via Accelerate vDSP runs in sub-millisecond time at N=200,000. LightGP compiles as a single static library with zero external dependencies and is installable via textttpip install lightgp

Key facts. No citing papers in the map yet (data through 7 Oct 2026). Builds on 5 of its 9 references that are in the map. First posted on arXiv 18 May 2026.

Builds on

5 of 9 references.

References · 5

  1. 2018GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Accelerationcited by 603
  2. 2015Kernel Interpolation for Scalable Structured Gaussian Processes (KISS-GP)cited by 175
  3. 2016GPflow: A Gaussian process library using TensorFlowcited by 139
  4. 2020Kernel Operations on the GPU, with Autodiff, without Memory Overflowscited by 101
  5. 2018Constant-Time Predictive Distributions for Gaussian Processescited by 48

Built on by

Not cited yet.