Cover Image

laplax -- Laplace Approximations with JAX

Fuente: arXiv

Saved in:

Bibliographic Details
Main Authors:	Weber, Tobias, Mucsányi, Bálint, Rommel, Lenard, Christie, Thomas, Kasüschke, Lars, Pförtner, Marvin, Hennig, Philipp
Format:	Preprint
Published:	2025
Subjects:	Machine Learning Artificial Intelligence
Online Access:	Acceder al recurso
Tags:	Add Tag No Tags, Be the first to tag this record!

Similar Items

FSP-Laplace: Function-Space Priors for the Laplace Approximation in Bayesian Deep Learning
by: Cinquin, Tristan, et al.
Published: (2024)

Rethinking Approximate Gaussian Inference in Classification
by: Mucsányi, Bálint, et al.
Published: (2025)

Geometric Gaussian Approximations of Probability Distributions
by: Da Costa, Nathaël, et al.
Published: (2025)

Linearization Turns Neural Operators into Function-Valued Gaussian Processes
by: Magnani, Emilia, et al.
Published: (2024)

Kronecker-factored Approximate Curvature (KFAC) From Scratch
by: Dangel, Felix, et al.
Published: (2025)

Flexible and Efficient Probabilistic PDE Solvers through Gaussian Markov Random Fields
by: Weiland, Tim, et al.
Published: (2025)

Scaling up Probabilistic PDE Simulators with Structured Volumetric Information
by: Weiland, Tim, et al.
Published: (2024)

Dynamics Reveals Structure: Challenging the Linear Propagation Assumption
by: Chang, Hoyeon, et al.
Published: (2026)

Debiasing Mini-Batch Quadratics for Applications in Deep Learning
by: Tatzel, Lukas, et al.
Published: (2024)

Skill Learning via Policy Diversity Yields Identifiable Representations for Reinforcement Learning
by: Reizinger, Patrik, et al.
Published: (2025)

Muon is Not That Special: Random or Inverted Spectra Work Just as Well
by: Shumaylov, Zakhar, et al.
Published: (2026)

Computation-Aware Kalman Filtering and Smoothing
by: Pförtner, Marvin, et al.
Published: (2024)

Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers
by: Pförtner, Marvin, et al.
Published: (2022)

On the Laplace Approximation as Model Selection Criterion for Gaussian Processes
by: Besginow, Andreas, et al.
Published: (2024)

Self-Supervised Laplace Approximation for Bayesian Uncertainty Quantification
by: Rodemann, Julian, et al.
Published: (2026)

Tubular Riemannian Laplace Approximations for Bayesian Neural Networks
by: David, Rodrigo Pereira
Published: (2025)

Sample Path Regularity of Gaussian Processes from the Covariance Kernel
by: Da Costa, Nathaël, et al.
Published: (2023)

CAX: Cellular Automata Accelerated in JAX
by: Faldor, Maxence, et al.
Published: (2024)

minimax: Efficient Baselines for Autocurricula in JAX
by: Jiang, Minqi, et al.
Published: (2023)

PuzzleJAX: A Benchmark for Reasoning and Learning
by: Earle, Sam, et al.
Published: (2025)

NAVIX: Scaling MiniGrid Environments with JAX
by: Pignatelli, Eduardo, et al.
Published: (2024)

Improving Group Fairness in Knowledge Distillation via Laplace Approximation of Early Exits
by: Fasth, Edvin, et al.
Published: (2025)

Chargax: A JAX Accelerated EV Charging Simulator
by: Ponse, Koen, et al.
Published: (2025)

Learning Bug Context for PyTorch-to-JAX Translation with LLMs
by: Phan, Hung, et al.
Published: (2025)

Jumanji: a Diverse Suite of Scalable Reinforcement Learning Environments in JAX
by: Bonnet, Clément, et al.
Published: (2023)

Mahjax: A GPU-Accelerated Mahjong Simulator for Reinforcement Learning in JAX
by: Nishimori, Soichiro, et al.
Published: (2026)

Achieving Approximate Symmetry Is Exponentially Easier than Exact Symmetry
by: Tahmasebi, Behrooz, et al.
Published: (2025)

Jasmine: A Simple, Performant and Scalable JAX-based World Modeling Codebase
by: Mahajan, Mihir, et al.
Published: (2025)

JaxARC: A High-Performance JAX-based Environment for Abstraction and Reasoning Research
by: Aadam, et al.
Published: (2026)

Scaling Is All You Need: Autonomous Driving with JAX-Accelerated Reinforcement Learning
by: Harmel, Moritz, et al.
Published: (2023)

Identity Curvature Laplace Approximation for Improved Out-of-Distribution Detection
by: Zhdanov, Maksim, et al.
Published: (2023)

JaxMARL: Multi-Agent RL Environments and Algorithms in JAX
by: Rutherford, Alexander, et al.
Published: (2023)

LAPLEX: The FFT of Learnable Laplace Kernels
by: Struski, Łukasz, et al.
Published: (2026)

Conditional Clifford-Steerable CNNs with Complete Kernel Basis for PDE Modeling
by: Szarvas, Bálint László, et al.
Published: (2025)

Towards Leveraging AutoML for Sustainable Deep Learning: A Multi-Objective HPO Approach on Deep Shift Neural Networks
by: Hennig, Leona, et al.
Published: (2024)

Benchmarking Uncertainty Disentanglement: Specialized Uncertainties for Specialized Tasks
by: Mucsányi, Bálint, et al.
Published: (2024)

Reparameterization invariance in approximate Bayesian inference
by: Roy, Hrittik, et al.
Published: (2024)

JPC: Flexible Inference for Predictive Coding Networks in JAX
by: Innocenti, Francesco, et al.
Published: (2024)

Latent Laplace Diffusion for Irregular Multivariate Time Series
by: You, Zinuo, et al.
Published: (2026)

Logit Reweighting for Topic-Focused Summarization
by: Braun, Joschka, et al.
Published: (2025)