Laplace approximations in JAX.
☆50Jun 16, 2026Updated last month
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☆101Oct 24, 2025Updated 9 months ago
- Style pack for the Tübingen AI Center, with templates for slides, theses, and more.☆20May 8, 2026Updated 2 months ago
- Bayesian inference on spatial and spatiotemporal data, faster than you can say "Cholesky!"☆18Jul 17, 2026Updated last week
- Integrals of Gaussians under linear domain constraints☆15Feb 27, 2020Updated 6 years ago
- Matrix-free linear algebra in JAX.☆178Jul 1, 2026Updated 3 weeks ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- 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, ...)☆69Jul 17, 2026Updated last week
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆35Apr 26, 2024Updated 2 years 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.☆71May 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…☆64Jul 9, 2026Updated 2 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
- 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.
- Code for the ICML 2026 Tutorial "Probabilistic Numerics — Computation is Machine Learning"☆56Jul 6, 2026Updated 2 weeks ago
- [🔥ICLR26 Oral] RealPDEBench: A Benchmark for Complex Physical Systems with Paired Real-World and Simulated Data☆103Jul 16, 2026Updated last week
- Repository of the journal club "Diffusion Models and Generative Modeling"☆16Nov 20, 2024Updated last year
- Training methodologies for autoregressive neural operators/emulators in JAX.☆13Jun 1, 2026Updated last month
- Probabilistic Numerics in Python.☆461Jul 3, 2025Updated last year
- State-space model inference with JAX☆71Updated this week
- Efficient Differentiable n-d PDE Solvers in JAX.☆215Jun 1, 2026Updated last month
- Code accompanying the latent-action-priors paper.☆12Mar 5, 2025Updated last year
- Gaussian processes with spherical harmonic features in JAX☆16Aug 24, 2025Updated 11 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.
- Step size adaptation for the No-U-Turn Sampler☆28Nov 25, 2025Updated 8 months ago
- Figure sizes, font sizes, fonts, and more configurations at minimal overhead. Fix your journal papers, conference proceedings, and other …☆749Mar 24, 2026Updated 4 months ago
- ☆16Apr 8, 2025Updated last year
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆232Mar 29, 2024Updated 2 years ago
- Neural Emulator Architectures in JAX.☆25Jun 1, 2026Updated last month
- IterGP: Computation-Aware Gaussian Process Inference (NeurIPS 2022)☆43Apr 12, 2023Updated 3 years ago
- Inference Combinators in JAX☆53May 17, 2025Updated last year
- A simple library to run variational inference on Stan models.☆33Apr 11, 2023Updated 3 years ago
- Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks☆11Oct 21, 2022Updated 3 years ago
- AI Agents on DigitalOcean Gradient AI Platform • AdBuild production-ready AI agents using customizable tools or access multiple LLMs through a single endpoint. Create custom knowledge bases or connect external data.
- Codes for Hilbert space reduced-rank GP regression☆16Jul 30, 2019Updated 6 years ago
- An extension of NumPyro for first-class support of dynamical systems.☆47Updated this week
- Julia interface to MNE-Python via PythonCall☆30Updated this week
- Code for the paper "Randomly pivoted Cholesky: Practical approximation of a kernel matrix with few entry evaluations"☆36Dec 4, 2025Updated 7 months ago
- ☆16Apr 26, 2023Updated 3 years 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 8 months ago