Laplace approximations in JAX.
☆50Jun 16, 2026Updated 2 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☆101Oct 24, 2025Updated 10 months ago
- Bayesian inference on spatial and spatiotemporal data, faster than you can say "Cholesky!"☆20Aug 23, 2026Updated last week
- 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.☆180Jul 1, 2026Updated 2 months ago
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- A package for computing matrix exponentials and finite horizon Gramians☆11Jan 21, 2026Updated 7 months ago
- 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 last month
- Code for the Paper "Physics-Informed Gaussian Process Regression Generalizes Linear PDE Solvers"☆36Apr 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
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Jul 19, 2023Updated 3 years ago
- ☆18Sep 3, 2024Updated 2 years ago
- 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…☆65Updated this week
- GPUs on demand by Runpod - Special Offer Available • AdRun AI, ML, and HPC workloads on powerful cloud GPUs—without limits or wasted spend. Deploy GPUs in under a minute and pay by the second.
- 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.☆16Sep 3, 2024Updated 2 years ago
- Code for the ICML 2026 Tutorial "Probabilistic Numerics — Computation is Machine Learning"☆61Jul 6, 2026Updated last month
- [🔥ICLR26 Oral] RealPDEBench: A Benchmark for Complex Physical Systems with Paired Real-World and Simulated Data☆118Jul 16, 2026Updated last month
- 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.☆460Jul 3, 2025Updated last year
- State-space model inference with JAX☆72Updated this week
- 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.
- Efficient Differentiable n-d PDE Solvers in JAX.☆231Jun 1, 2026Updated 3 months ago
- Gaussian processes with spherical harmonic features in JAX☆16Aug 24, 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 …☆755Mar 24, 2026Updated 5 months ago
- 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.☆27Aug 25, 2026Updated last week
- IterGP: Computation-Aware Gaussian Process Inference (NeurIPS 2022)☆44Apr 12, 2023Updated 3 years 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
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- Coordinates in JAX☆37Updated 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
- Probabilistic programming for dynamical systems! Supports models with NumPyro, deterministic & stochastic systems, discrete- and continuo…☆57Updated this week
- Julia interface to MNE-Python via PythonCall☆30Aug 25, 2026Updated last week
- Code for the paper "Randomly pivoted Cholesky: Practical approximation of a kernel matrix with few entry evaluations"☆38Aug 20, 2026Updated 2 weeks ago
- ☆16Apr 26, 2023Updated 3 years ago