Laplace approximations in JAX.
☆52Jun 16, 2026Updated 3 months ago
Alternatives and similar repositories for laplax
Users that are interested in laplax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Style pack for the Tübingen AI Center, with templates for slides, theses, and more.☆24Sep 4, 2026Updated 2 weeks ago
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆15May 8, 2023Updated 3 years ago
- Integrals of Gaussians under linear domain constraints☆15Feb 27, 2020Updated 6 years ago
- Matrix-free linear algebra in JAX.☆180Sep 2, 2026Updated 3 weeks ago
- Code accompanying the NeurIPS 2021 Paper: A Probabilistic State Space Model for Joint Inference from Differential Equations and Data (Sch…☆13Nov 7, 2022Updated 3 years ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulati…☆37Jul 21, 2025Updated last year
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆71Jul 17, 2026Updated 2 months ago
- Probabilistic ODE solvers are fun, but are they fast? See also: https://github.com/pnkraemer/probdiffeq for JAX code or https://github.c…☆22Jul 20, 2024Updated 2 years ago
- Large-scale uncertainty benchmark in deep learning.☆70May 10, 2025Updated last year
- Probabilistic Finite Volume Method based on Affine Gaussian Process inference☆11Jun 10, 2024Updated 2 years ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆65Sep 2, 2026Updated 3 weeks ago
- Finite State Machine implementations of MCMC algorithms in JAX for efficient vectorization.☆20Jun 7, 2025Updated last year
- Computation-Aware Kalman Filtering and RTS Smoothing☆20Mar 6, 2025Updated last year
- An implementation of the differentially private variational inference algorithm for NumPyro.☆17Sep 3, 2024Updated 2 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- Code for the ICML 2026 Tutorial "Probabilistic Numerics — Computation is Machine Learning"☆61Jul 6, 2026Updated 2 months ago
- [🔥ICLR26 Oral] RealPDEBench: A Benchmark for Complex Physical Systems with Paired Real-World and Simulated Data☆126Jul 16, 2026Updated 2 months ago
- Repository of the journal club "Diffusion Models and Generative Modeling"☆15Nov 20, 2024Updated last year
- Training methodologies for autoregressive neural operators/emulators in JAX.☆15Jun 1, 2026Updated 3 months ago
- Probabilistic Numerics in Python.☆462Jul 3, 2025Updated last year
- Efficient Differentiable n-d PDE Solvers in JAX.☆231Jun 1, 2026Updated 3 months ago
- Code accompanying the latent-action-priors paper.☆12Mar 5, 2025Updated last year
- Step size adaptation for the No-U-Turn Sampler☆29Nov 25, 2025Updated 9 months ago
- Figure sizes, font sizes, fonts, and more configurations at minimal overhead. Fix your journal papers, conference proceedings, and other …☆754Mar 24, 2026Updated 6 months ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- ☆16Apr 8, 2025Updated last year
- IterGP: Computation-Aware Gaussian Process Inference (NeurIPS 2022)☆44Apr 12, 2023Updated 3 years ago
- Neural Emulator Architectures in JAX.☆27Aug 25, 2026Updated 3 weeks ago
- Inference Combinators in JAX☆54May 17, 2025Updated last year
- A simple library to run variational inference on Stan models.☆33Apr 11, 2023Updated 3 years ago
- Coordinates in JAX☆38Updated this week
- Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks☆11Oct 21, 2022Updated 3 years ago
- Codes for Hilbert space reduced-rank GP regression☆15Jul 30, 2019Updated 7 years ago
- Julia interface to MNE-Python via PythonCall☆30Sep 8, 2026Updated 2 weeks ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- Code for the paper "Randomly pivoted Cholesky: Practical approximation of a kernel matrix with few entry evaluations"☆38Aug 20, 2026Updated last month
- Probabilistic programming for dynamical systems! Supports models with NumPyro, deterministic & stochastic systems, discrete- and continuo…☆78Updated this week
- A generic library for linear and non-linear Gaussian smoothing problems. The code leverages JAX and implements several linearization algo…☆15Apr 20, 2026Updated 5 months ago
- Code for "Symplectic Adjoint Method for Exact Gradient of Neural ODE with Minimal Memory," NeurIPS, 2021.☆18Jan 8, 2022Updated 4 years ago
- A repository for Mixed Precision Training in JAX based on Equinox.☆16Nov 2, 2025Updated 10 months ago
- Tool for exploring probability distributions.☆24Dec 18, 2024Updated last year
- Laplace Redux -- Effortless Bayesian Deep Learning☆46Jun 6, 2025Updated last year