GPyTorch: Blackbox Matrix-Matrix Gaussian Process Inference with GPU Acceleration
Not on the map. Why?
Abstract (arXiv)
Despite advances in scalable models, the inference tools used for Gaussian processes (GPs) have yet to fully capitalize on developments in computing hardware. We present an efficient and general approach to GP inference based on Blackbox Matrix-Matrix multiplication (BBMM). BBMM inference uses a modified batched version of the conjugate gradients algorithm to derive all terms for training and inference in a single call. BBMM reduces the asymptotic complexity of exact GP inference from O(n3) to O(n2). Adapting this algorithm to scalable approximations and complex GP models simply requires a routine for efficient matrix-matrix multiplication with the kernel and its derivative. In addition, BBMM uses a specialized preconditioner to substantially speed up convergence. In experiments we show that BBMM effectively uses GPU hardware to dramatically accelerate both exact GP inference and scalable approximations. Additionally, we provide GPyTorch, a software platform for scalable GP inference via BBMM, built on PyTorch.
Key facts. Cited by 609 papers in the map (data through 14 Oct 2026). First posted on arXiv 28 Sep 2018.
Builds on
No reference list yet.
Built on by
Citing papers per year: 2021: 87; 2022: 109; 2023: 103; 2024: 113; 2025: 110; 2026: 87.
609 papers.
Sort:
By area:
- 2022Scientific Machine Learning through Physics-Informed Neural Networks: Where we are and What's nextcited by 608
- 2021CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputationcited by 582
- 2023Large Language Models Are Zero-Shot Time Series Forecasterscited by 500
- 2021Hyperparameter Optimization: Foundations, Algorithms, Best Practices and Open Challengescited by 145
- 2022Transformer Neural Processes: Uncertainty-Aware Meta Learning Via Sequence Modelingcited by 127
- 2025TiRex: Zero-Shot Forecasting Across Long and Short Horizons with Enhanced In-Context Learningcited by 120
- 2022Accelerating Bayesian Optimization for Biological Sequence Design with Denoising Autoencoderscited by 102
- 2021On Feature Collapse and Deep Kernel Learning for Single Forward Pass Uncertaintycited by 87
- 2022Local Latent Space Bayesian Optimization over Structured Inputscited by 79
- 2023TabR: Tabular Deep Learning Meets Nearest Neighbors in 2023cited by 74
- 2023PFNs4BO: In-Context Learning for Bayesian Optimizationcited by 73
- 2022Deep Ensembles Work, But Are They Necessary?cited by 63
38 more
- 2023Black Box Adversarial Prompting for Foundation Modelscited by 62
- 2023Geographic Location Encoding with Spherical Harmonics and Sinusoidal Representation Networkscited by 61
- 2022Joint Entropy Search for Maximally-Informed Bayesian Optimizationcited by 55
- 2021Think Global and Act Local: Bayesian Optimisation over High-Dimensional Categorical and Mixed Search Spacescited by 53
- 2021Rectangular Flows for Manifold Learningcited by 50
- 2021safe-control-gym: a Unified Benchmark Suite for Safe Learning-based Control and Reinforcement Learning in Roboticscited by 50
- 2022On Uncertainty, Tempering, and Data Augmentation in Bayesian Classificationcited by 46
- 2021Combining Latent Space and Structured Kernels for Bayesian Optimization over Combinatorial Spacescited by 45
- 2022Joint Entropy Search for Multi-objective Bayesian Optimizationcited by 45
- 2022GAUCHE: A Library for Gaussian Processes in Chemistrycited by 41
- 2021Local policy search with Bayesian optimizationcited by 38
- 2021Mosaic Flows: A Transferable Deep Learning Framework for Solving PDEs on Unseen Domainscited by 37
- 2021DOCKSTRING: easy molecular docking yields better benchmarks for ligand designcited by 35
- 2022Active Bayesian Causal Inferencecited by 35
- 2024A review on data-driven constitutive laws for solidscited by 35
- 2024Standard Gaussian Process is All You Need for High-Dimensional Bayesian Optimizationcited by 34
- 2022NeuralUQ: A comprehensive library for uncertainty quantification in neural differential equations and operatorscited by 33
- 2021Fast Adaptation with Linearized Neural Networkscited by 32
- 2021Bayesian Optimization with High-Dimensional Outputscited by 32
- 2021A Metalearning Approach for Physics-Informed Neural Networks (PINNs): Application to Parameterized PDEscited by 32
- 2021Spatio-Temporal Variational Gaussian Processescited by 32
- 2022Discovering Many Diverse Solutions with Bayesian Optimizationcited by 31
- 2023RLSbench: Domain Adaptation Under Relaxed Label Shiftcited by 30
- 2021Valid prediction intervals for regression problemscited by 29
- 2022Bayesian Optimization with Conformal Prediction Setscited by 28
- 2023Bounce: Reliable High-Dimensional Bayesian Optimization for Combinatorial and Mixed Spacescited by 28
- 2022Disentangling Epistemic and Aleatoric Uncertainty in Reinforcement Learningcited by 27
- 2023Sampling from Gaussian Process Posteriors using Stochastic Gradient Descentcited by 27
- 2023Looping in the Human Collaborative and Explainable Bayesian Optimizationcited by 27
- 2022Bayesian Active Learning with Fully Bayesian Gaussian Processescited by 26
- 2022Making Look-Ahead Active Learning Strategies Feasible with Neural Tangent Kernelscited by 25
- 2022Exploring Predictive Uncertainty and Calibration in NLP: A Study on the Impact of Method & Data Scarcitycited by 25
- 2023ClimateSet: A Large-Scale Climate Model Dataset for Machine Learningcited by 25
- 2021Deconditional Downscaling with Gaussian Processescited by 24
- 2021Geometry-aware Bayesian Optimization in Robotics using Riemannian Mat'ern Kernelscited by 24
- 2022Posterior and Computational Uncertainty in Gaussian Processescited by 24
- 2023Bayesian Optimization over High-Dimensional Combinatorial Spaces via Dictionary-based Embeddingscited by 24
- 2021Meta-Learning Reliable Priors in the Function Spacecited by 23
50 most cited of 609 All 609 (JSON)
- 2026GpFlash: Maximizing Speed and Minimizing Memory for Gaussian Process Training and Prediction12 Oct 2026
- 2026Ghost tasking for parametrized Gaussian Processes solving linear differential equations8 Oct 2026
- 2026Isotropic Gaussian Processes Improve Vanilla Bayesian Optimization in High Dimensions5 Oct 2026
- 2026Improving Function Space Flow Matching with Kernel Optimal Transport29 Sep 2026
- 2026Sparsity by Default: The Theory and Practice of ARD in Gaussian Process Regression for Variable Selection27 Sep 2026
- 2026Learning What to Evaluate: Correlation-Aware Decoupling for Multiobjective Bayesian Optimization26 Sep 2026
- 2026In-Context Guidance: Learning Inter-Task Synergies via Numerical Foundational Models for Few-Shot Multitask Optimization22 Sep 2026
- 2026Triply-Scalable Equivariant Gaussian Process Modeling17 Sep 2026
- 2026CrystalMO-TuRBO: Multi-Objective Trust-Region Bayesian Optimization for High-precision Joint Crystal Structure Refinement17 Sep 2026
- 2026Portfolio-Based Constrained Multi-Objective Bayesian Optimization for Materials Design17 Sep 2026
- 2026Autonomous Research Project Management as an Agent Skill: A Case Study in Exact Spectral Spatial Regression16 Sep 2026
- 2026Kernel-Based Metrics Learning for Uncertain Opponent Vehicle Trajectory Prediction in Autonomous Racing15 Sep 2026
38 more
- 2026Bayesian Optimisation Using Product-of-Experts Gaussian Process Models with Uncertainty Calibration14 Sep 2026
- 2026Converge Then Diversify: Decoupling Convergence and Diversity in Multi-Objective Bayesian Optimisation11 Sep 2026
- 2026Dynamic language model representations for multi-objective reaction optimisation10 Sep 2026
- 2026GPU-Enabled Large-Scale Optimization Using Randomized Linear Algebra8 Sep 2026
- 2026Information-Based Calibration of Uncertainty Quantification in Product-of-Experts Gaussian Process Models29 Aug 2026
- 2026Why not to use the Gaussian kernel27 Aug 2026
- 2026GRAPE: Gradient Refinement and Progress-Aware Exploitation for Query-Efficient High-Dimensional Bayesian Optimization25 Aug 2026
- 2026On the Brittleness of Maximum Likelihood Estimation for Gaussian Process Hyperparameter Optimization13 Aug 2026
- 2026Finding the Needle in a Haystack: Test-Time Analog Circuit Representation Adaptation for Bayesian Optimization13 Aug 2026
- 2026Agentic Bayesian Optimization through Surrogate-Augmented Autoresearch31 Jul 2026
- 2026Variational Quantum Conditional Boltzmann Machines for Time-Series Forecasting: Architectures, Symmetric Hyperparameter Evaluation, and a Nonlinear Benchmark27 Jul 2026
- 2026Deep Sigma Point Processes for RCS Modeling in Spaceborne SAR Imagery23 Jul 2026
- 2026Maximally Robust Satisficing Bayesian Optimization15 Jul 2026
- 2026Pitfalls and Remedies for Multi-Task Bayesian Optimization10 Jul 2026
- 2026Koopman operator theory: fundamentals, control, and applications2 Jul 2026
- 2026Online Data Selection for Instruction Tuning via Gaussian Processes29 Jun 2026
- 2026Surrogate-Gated Generation and Foundation-Model Embeddings for Bayesian Materials Design26 Jun 2026
- 2026Decision-Aligned Evaluation of Uncertainty Quantification25 Jun 2026
- 2026Interpretable Probabilistic Medical Image Segmentation via Gaussian Process with Explicit Modelling of Annotation Bias and Variability22 Jun 2026
- 2026REMAL: Residual Equilibrium Manifold Active Learning for Surrogate-Based Multidisciplinary Design Analysis11 Jun 2026
- 2026A green solvent screening tool for emerging materials via uncertainty aware, transformer enhanced transfer learning11 Jun 2026
- 2026PolicyGuard: Towards Test-time and Step-level Adversary (Backdoor) Defense for Reinforcement Learning Agent11 Jun 2026
- 2026Gaussian Process Latent Factor Regression for Low-Data, High-Dimensional Output Problems4 Jun 2026
- 2026U-Net-Accelerated Quality-Diversity Optimization for Climate-Adaptive Urban Layouts3 Jun 2026
- 2026Scalable Derivative Gaussian Processes via Exact Gradient Reduction1 Jun 2026
- 2026ShaplEIG: Bayesian Experimental Design for Shapley Value Estimation1 Jun 2026
- 2026Revisiting Neural Processes via Fourier Transform and Volterra Series31 May 2026
- 2026Joint Model and Data Sparsification via the Marginal Likelihood28 May 2026
- 2026Supervised Distributional Reduction via Optimal Transport and Dependence Maximization26 May 2026
- 2026Probabilistic Data-Driven Modelling of Astrophysical Transients: The Neural Process Family for Ultrafast and Class-Agnostic Light Curve Reconstruction with NightLANP26 May 2026
- 2026SIKA-GP: Accelerating Gaussian Process Inference with Sparse Inducing Kernel Approximations for Bayesian Deep Learning26 May 2026
- 2026Lightweight Gaussian Process Inference in C++ on Metal and CUDA18 May 2026
- 2026Concepts Worth Having: Refining VLM-Guided Concept Bottleneck Models with Minimal Annotations13 May 2026
- 2026IV-ICL: Bounding Causal Effects with Instrumental Variables via In-Context Learning13 May 2026
- 2026Scalable Gaussian process inference via neural feature maps11 May 2026
- 2026Implicit Multi-Camera System Calibration Using Gaussian Processes8 May 2026
- 2026Transformed Latent Variable Multi-Output Gaussian Processes6 May 2026
- 2026A Target-Free Harmonization Method for MRI2 May 2026
50 newest of 609 All 609 (JSON)
- 2024A Safe and Data-efficient Model-based Reinforcement Learning System for HVAC Controlcited by 5
- 2024Multi-Fidelity Reinforcement Learning for Time-Optimal Quadrotor Re-planningcited by 4
- 2025No-Regret Thompson Sampling for Finite-Horizon Markov Decision Processes with Gaussian Processescited by 4
- 2024Efficient Model-Based Reinforcement Learning Through Optimistic Thompson Samplingcited by 3
- 2026Structure Detection for Contextual Reinforcement Learningcited by 2
- 2024Autonomous Wheel Loader Navigation Using Goal-Conditioned Actor-Critic MPCcited by 0
- 2026Artificial Intelligence for Climate Adaptation: Reinforcement Learning for Climate Change-Resilient Transportcited by 0
- 2026Artificial Intelligence for Climate Adaptation: Reinforcement Learning for Climate Change-Resilient Transport6 Mar 2026
- 2026Structure Detection for Contextual Reinforcement Learning13 Jan 2026
- 2025No-Regret Thompson Sampling for Finite-Horizon Markov Decision Processes with Gaussian Processes23 Oct 2025
- 2024Efficient Model-Based Reinforcement Learning Through Optimistic Thompson Sampling7 Oct 2024
- 2024Autonomous Wheel Loader Navigation Using Goal-Conditioned Actor-Critic MPC24 Sep 2024
- 2024A Safe and Data-efficient Model-based Reinforcement Learning System for HVAC Control16 Jul 2024
- 2024Multi-Fidelity Reinforcement Learning for Time-Optimal Quadrotor Re-planning13 Mar 2024
- 2021CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputationcited by 582
- 2024Sparse Inducing Points in Deep Gaussian Processes: Enhancing Modeling with Denoising Diffusion Variational Inferencecited by 16
- 2025Residual-augmented flow matching operators for probabilistic partial differential equationscited by 2
- 2024Multiple-Source Localization from a Single-Snapshot Observation Using Graph Bayesian Optimizationcited by 0
- 2025Generative AI Against Poaching: Latent Composite Flow Matching for Wildlife Conservationcited by 0
- 2026Improving Function Space Flow Matching with Kernel Optimal Transportcited by 0
- 2026Improving Function Space Flow Matching with Kernel Optimal Transport29 Sep 2026
- 2025Residual-augmented flow matching operators for probabilistic partial differential equations14 Dec 2025
- 2025Generative AI Against Poaching: Latent Composite Flow Matching for Wildlife Conservation20 Aug 2025
- 2024Sparse Inducing Points in Deep Gaussian Processes: Enhancing Modeling with Denoising Diffusion Variational Inference24 Jul 2024
- 2024Multiple-Source Localization from a Single-Snapshot Observation Using Graph Bayesian Optimization25 Mar 2024
- 2021CSDI: Conditional Score-based Diffusion Models for Probabilistic Time Series Imputation7 Jul 2021
- 2023Black Box Adversarial Prompting for Foundation Modelscited by 62
- 2025Is Multilingual LLM Watermarking Truly Multilingual? Scaling Robustness to 100+ Languages via Back-Translationcited by 1
- 2026Safety-Efficacy Trade Off: Robustness against Data-Poisoningcited by 1
- 2026PolicyGuard: Towards Test-time and Step-level Adversary (Backdoor) Defense for Reinforcement Learning Agentcited by 1
- 2026PolicyGuard: Towards Test-time and Step-level Adversary (Backdoor) Defense for Reinforcement Learning Agent11 Jun 2026
- 2026Safety-Efficacy Trade Off: Robustness against Data-Poisoning31 Jan 2026
- 2025Is Multilingual LLM Watermarking Truly Multilingual? Scaling Robustness to 100+ Languages via Back-Translation20 Oct 2025
- 2023Black Box Adversarial Prompting for Foundation Models8 Feb 2023
- 2022Scientific Machine Learning through Physics-Informed Neural Networks: Where we are and What's nextcited by 608
- 2023Large Language Models Are Zero-Shot Time Series Forecasterscited by 500
- 2021Hyperparameter Optimization: Foundations, Algorithms, Best Practices and Open Challengescited by 145
- 2022Transformer Neural Processes: Uncertainty-Aware Meta Learning Via Sequence Modelingcited by 127
- 2025TiRex: Zero-Shot Forecasting Across Long and Short Horizons with Enhanced In-Context Learningcited by 120
- 2022Accelerating Bayesian Optimization for Biological Sequence Design with Denoising Autoencoderscited by 102
- 2021On Feature Collapse and Deep Kernel Learning for Single Forward Pass Uncertaintycited by 87
- 2022Local Latent Space Bayesian Optimization over Structured Inputscited by 79
- 2023TabR: Tabular Deep Learning Meets Nearest Neighbors in 2023cited by 74
- 2023PFNs4BO: In-Context Learning for Bayesian Optimizationcited by 73
- 2022Deep Ensembles Work, But Are They Necessary?cited by 63
- 2023Geographic Location Encoding with Spherical Harmonics and Sinusoidal Representation Networkscited by 61
12 most cited of 583 All 583 (JSON)
- 2026GpFlash: Maximizing Speed and Minimizing Memory for Gaussian Process Training and Prediction12 Oct 2026
- 2026Ghost tasking for parametrized Gaussian Processes solving linear differential equations8 Oct 2026
- 2026Isotropic Gaussian Processes Improve Vanilla Bayesian Optimization in High Dimensions5 Oct 2026
- 2026Sparsity by Default: The Theory and Practice of ARD in Gaussian Process Regression for Variable Selection27 Sep 2026
- 2026Learning What to Evaluate: Correlation-Aware Decoupling for Multiobjective Bayesian Optimization26 Sep 2026
- 2026In-Context Guidance: Learning Inter-Task Synergies via Numerical Foundational Models for Few-Shot Multitask Optimization22 Sep 2026
- 2026Triply-Scalable Equivariant Gaussian Process Modeling17 Sep 2026
- 2026CrystalMO-TuRBO: Multi-Objective Trust-Region Bayesian Optimization for High-precision Joint Crystal Structure Refinement17 Sep 2026
- 2026Portfolio-Based Constrained Multi-Objective Bayesian Optimization for Materials Design17 Sep 2026
- 2026Autonomous Research Project Management as an Agent Skill: A Case Study in Exact Spectral Spatial Regression16 Sep 2026
- 2026Bayesian Optimisation Using Product-of-Experts Gaussian Process Models with Uncertainty Calibration14 Sep 2026
- 2026Converge Then Diversify: Decoupling Convergence and Diversity in Multi-Objective Bayesian Optimisation11 Sep 2026
12 newest of 583 All 583 (JSON)