Kernel Operations on the GPU, with Autodiff, without Memory Overflows

27 March 2020

Benjamin Charlier

Jean Feydy

J. Glaunès

François-David Collin

G. Durif

ArXiv PDF HTML

Papers citing "Kernel Operations on the GPU, with Autodiff, without Memory Overflows"

29 / 29 papers shown

Title
Position: Curvature Matrices Should Be Democratized via Linear Operators Felix Dangel Runa Eschenhagen Weronika Ormaniec Andres Fernandez Lukas Tatzel Agustinus Kristiadi 58 3 0 31 Jan 2025
Fast Summation of Radial Kernels via QMC Slicing Johannes Hertrich Tim Jahn Michael Quellmalz 26 5 0 02 Oct 2024
Gradients of Functions of Large Matrices Nicholas Krämer Pablo Moreno-Muñoz Hrittik Roy Søren Hauberg 34 0 0 27 May 2024
Memory-Scalable and Simplified Functional Map Learning Robin Magnet M. Ovsjanikov 27 2 0 30 Mar 2024
Efficient and Accurate Optimal Transport with Mirror Descent and Conjugate Gradients Mete Kemertas Allan D. Jepson Amir-massoud Farahmand OT 58 3 0 17 Jul 2023
Toward Mesh-Invariant 3D Generative Deep Learning with Geometric Measures T. Besnier Sylvain Arguillere E. Pierson Mohamed Daoudi 3DH 16 8 0 27 Jun 2023
Properties of Marginal Sequential Monte Carlo Methods F. R. Crucinio A. M. Johansen 26 2 0 06 Mar 2023
Heterogeneous reconstruction of deformable atomic models in Cryo-EM Y. Nashed A. Peck Julien N. P. Martel A. Levy Bongjin Koo Gordon Wetzstein Nina Miolane Daniel Ratner Frédéric Poitevin 13 10 0 29 Sep 2022
Liquid Structural State-Space Models Ramin Hasani Mathias Lechner Tsun-Hsuan Wang Makram Chahine Alexander Amini Daniela Rus AI4TS 101 95 0 26 Sep 2022
StreamNet: A WAE for White Matter Streamline Analysis Andrew Lizarraga K. Narr Kristy A. Donald S. H. Joshi 22 4 0 03 Sep 2022
Memory Safe Computations with XLA Compiler A. Artemev Tilman Roeder Mark van der Wilk 16 8 0 28 Jun 2022
Diffeomorphic Registration using Sinkhorn Divergences Lucas de Lara Alberto González Sanz Jean-Michel Loubes FedML 13 8 0 28 Jun 2022
Fast Nonlinear Vector Quantile Regression Aviv A. Rosenberg S. Vedula Yaniv Romano A. Bronstein 18 7 0 30 May 2022
FlashAttention: Fast and Memory-Efficient Exact Attention with IO-Awareness Tri Dao Daniel Y. Fu Stefano Ermon Atri Rudra Christopher Ré VLM 58 2,023 0 27 May 2022
How to Register a Live onto a Liver ? Partial Matching in the Space of Varifolds Pierre-Louis Antonsanti T. Benseghir V. Jugnon M. Ghosn P. Chassat Irene Kaltenmark J. Glaunès 17 0 0 12 Apr 2022
Diagonal State Spaces are as Effective as Structured State Spaces Ankit Gupta Albert Gu Jonathan Berant 37 290 0 27 Mar 2022
Giga-scale Kernel Matrix Vector Multiplication on GPU Robert Hu Siu Lun Chau Dino Sejdinovic J. Glaunès 21 2 0 02 Feb 2022
Sinkformers: Transformers with Doubly Stochastic Attention Michael E. Sander Pierre Ablin Mathieu Blondel Gabriel Peyré 29 76 0 22 Oct 2021
Unrolled Variational Bayesian Algorithm for Image Blind Deconvolution Yunshi Huang Émilie Chouzenoux J. Pesquet BDL 11 12 0 14 Oct 2021
Nonlinearities in Steerable SO(2)-Equivariant CNNs Daniel Franzen Michael Wand 33 3 0 14 Sep 2021
Learning Partial Differential Equations in Reproducing Kernel Hilbert Spaces George Stepaniants 46 15 0 26 Aug 2021
Optimizing full 3D SPARKLING trajectories for high-resolution T2*-weighted Magnetic Resonance Imaging R. ChaithyaG. P. Weiss Guillaume Daval-Frérot Aurélien Massire A. Vignaud P. Ciuciu 6 8 0 06 Aug 2021
Fast and Scalable Optimal Transport for Brain Tractograms Jean Feydy Pierre Roussillon A. Trouvé Pietro Gori OT 19 24 0 05 Jul 2021
Preconditioning for Scalable Gaussian Process Hyperparameter Optimization Jonathan Wenger Geoff Pleiss Philipp Hennig John P. Cunningham J. Gardner 14 23 0 01 Jul 2021
Shape registration in the time of transformers Giovanni Trappolini Luca Cosmo Luca Moschella R. Marin Simone Melzi Emanuele Rodolà ViT 3DPC MedIm 30 56 0 25 Jun 2021
Tighter Bounds on the Log Marginal Likelihood of Gaussian Process Regression Using Conjugate Gradients A. Artemev David R. Burt Mark van der Wilk 18 18 0 16 Feb 2021
Neural Splines: Fitting 3D Surfaces with Infinitely-Wide Neural Networks Francis Williams Matthew Trager Joan Bruna Denis Zorin 16 67 0 24 Jun 2020
Fast Matrix Square Roots with Applications to Gaussian Processes and Bayesian Optimization Geoff Pleiss M. Jankowiak David Eriksson Anil Damle J. Gardner 17 43 0 19 Jun 2020
Sinkhorn Divergences for Unbalanced Optimal Transport Thibault Séjourné Jean Feydy Franccois-Xavier Vialard A. Trouvé Gabriel Peyré OT 19 71 0 28 Oct 2019