Laplace approximations in JAX.
☆52Jun 16, 2026Updated 4 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:
- Sketched linear operations for PyTorch☆102Oct 24, 2025Updated last year
- Probabilistic numerical finite differences. Compute finite difference weights and differentiation matrices on scattered data sites and wi…☆15May 8, 2023Updated 3 years ago
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆37Apr 26, 2024Updated 2 years ago
- Bencher makes evaluating black-box problems simple. See https://arxiv.org/abs/2505.21321☆14Oct 1, 2026Updated last month
- Matrix-free linear algebra in JAX.☆183Oct 22, 2026Updated last week
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- Large-scale uncertainty benchmark in deep learning.☆70May 10, 2025Updated last year
- This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulati…☆37Jul 21, 2025Updated last year
- Bayesian inference on spatial and spatiotemporal data, faster than you can say "Cholesky!"☆30Oct 25, 2026Updated last week
- 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
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆71Jul 17, 2026Updated 3 months ago
- ☆18Sep 3, 2024Updated 2 years ago
- Probabilistic Finite Volume Method based on Affine Gaussian Process inference☆11Jun 10, 2024Updated 2 years ago
- Gaussian processes with spherical harmonic features in JAX☆16Aug 24, 2025Updated last year
- Bayesian Modelling using Jax☆17Updated this week
- End-to-end encrypted email - Proton Mail • AdSpecial offer: 40% Off Yearly / 80% Off First Month. All Proton services are open source and independently audited for security.
- Training methodologies for autoregressive neural operators/emulators in JAX.☆16Jun 1, 2026Updated 5 months ago
- Finite State Machine implementations of MCMC algorithms in JAX for efficient vectorization.☆21Jun 7, 2025Updated last year
- A simple library to run variational inference on Stan models.☆33Apr 11, 2023Updated 3 years ago
- Code for the ICML 2026 Tutorial "Probabilistic Numerics — Computation is Machine Learning"☆64Jul 6, 2026Updated 3 months ago
- Probabilistic solvers for differential equations in JAX. Adaptive ODE solvers with calibration, state-space model factorisations, and cus…☆66Oct 2, 2026Updated last month
- Lightning-like training API for JAX with Flax☆45Dec 8, 2024Updated last year
- An implementation of the differentially private variational inference algorithm for NumPyro.☆17Sep 3, 2024Updated 2 years ago
- A curated list of resources for learning Gaussian Processes☆41Jul 18, 2021Updated 5 years ago
- Probabilistic Numerics in Python.☆463Jul 3, 2025Updated last year
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆26Nov 18, 2025Updated 11 months ago
- Laplace Redux -- Effortless Bayesian Deep Learning☆47Jun 6, 2025Updated last year
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆234Mar 29, 2024Updated 2 years ago
- Step size adaptation for the No-U-Turn Sampler☆29Nov 25, 2025Updated 11 months ago
- ☆246Oct 23, 2026Updated last week
- Repository for NeurIPS 2025 paper, "Physics-informed Reduced Order Modeling of Time-dependent PDEs via Differentiable Solvers."☆11Nov 25, 2025Updated 11 months ago
- Code accompanying the latent-action-priors paper.☆12Mar 5, 2025Updated last year
- Finite difference tools in JAX☆23Feb 4, 2024Updated 2 years ago
- Efficient Differentiable n-d PDE Solvers in JAX.☆233Updated this week
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Figure sizes, font sizes, fonts, and more configurations at minimal overhead. Fix your journal papers, conference proceedings, and other …☆759Oct 12, 2026Updated 3 weeks ago
- [🔥ICLR26 Oral] RealPDEBench: A Benchmark for Complex Physical Systems with Paired Real-World and Simulated Data☆136Jul 16, 2026Updated 3 months ago
- Laplace approximations for Deep Learning.☆552Updated this week
- Reinforcement learning agents that can beat a human in a game of laser hockey. Team project on the course Reinforcement Learning WS 20/21…☆10Mar 18, 2021Updated 5 years ago
- Distrax, but in equinox. Lightweight JAX library of probability distributions and bijectors.☆48Sep 10, 2026Updated last month
- Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks☆11Oct 21, 2022Updated 4 years ago
- Gaussian processes in JAX and Equinox.☆669Updated this week