Laplace approximations in JAX.
☆49Jun 16, 2026Updated 2 weeks 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 8 months ago
- Bayesian inference on spatial and spatiotemporal data, faster than you can say "Cholesky!"☆15Jun 24, 2026Updated last week
- Integrals of Gaussians under linear domain constraints☆15Feb 27, 2020Updated 6 years ago
- This code accompanies the manuscript "A Generative Framework for Probabilistic, Spatiotemporally Coherent Downscaling of Climate Simulati…☆37Jul 21, 2025Updated 11 months 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
- 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.
- PyTorch linear operators for curvature matrices (Hessian, Fisher/GGN, KFAC, ...)☆68Updated this week
- A package for computing matrix exponentials and finite horizon Gramians☆11Jan 21, 2026Updated 5 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…☆21Jul 20, 2024Updated last year
- [TMLR 2022] Curvature access through the generalized Gauss-Newton's low-rank structure: Eigenvalues, eigenvectors, directional derivative…☆17Jul 19, 2023Updated 2 years ago
- ☆18Sep 3, 2024Updated last year
- Probabilistic Finite Volume Method based on Affine Gaussian Process inference☆11Jun 10, 2024Updated 2 years ago
- Large-scale uncertainty benchmark in deep learning.☆71May 10, 2025Updated last year
- An implementation of the differentially private variational inference algorithm for NumPyro.☆16Sep 3, 2024Updated last year
- Computation-Aware Kalman Filtering and RTS Smoothing☆19Mar 6, 2025Updated last year
- 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.
- Repository of the journal club "Diffusion Models and Generative Modeling"☆16Nov 20, 2024Updated last year
- Code accompanying the latent-action-priors paper.☆12Mar 5, 2025Updated last year
- Figure sizes, font sizes, fonts, and more configurations at minimal overhead. Fix your journal papers, conference proceedings, and other …☆745Mar 24, 2026Updated 3 months ago
- IterGP: Computation-Aware Gaussian Process Inference (NeurIPS 2022)☆43Apr 12, 2023Updated 3 years ago
- Posterior Refinement Improves Sample Efficiency in Bayesian Neural Networks☆11Oct 21, 2022Updated 3 years ago
- Probabilistic Numerics in Python.☆461Jul 3, 2025Updated last year
- ☆16Apr 26, 2023Updated 3 years ago
- Notes for the Numerics of Machine Learning Lecture Course at the University of Tübingen☆230Mar 29, 2024Updated 2 years ago
- Code for "A survey and benchmark of high-dimensional Bayesian optimization of discrete sequences"☆16Feb 24, 2025Updated last year
- 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.
- Modular Gaussian Processes☆17Dec 9, 2021Updated 4 years ago
- the indexer and search engine for irchiver, see https://irchiver.com for license and other information☆15Dec 2, 2021Updated 4 years ago
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆25Nov 18, 2025Updated 7 months ago
- Matrix-free linear algebra in JAX.☆174Jun 23, 2026Updated last week
- Materials for a training workshop on 'Introduction to Julia for Computational Science'☆12Jul 9, 2025Updated 11 months ago
- Marginal distributions and Markov kernels that play nice with each other for the purpose of Bayesian state estimation.☆18Oct 22, 2025Updated 8 months ago
- ☆16May 2, 2026Updated 2 months ago
- [ICML 2024] SINGD: KFAC-like Structured Inverse-Free Natural Gradient Descent (http://arxiv.org/abs/2312.05705)☆24Nov 4, 2024Updated last year
- A library of discrete objectives☆27Oct 5, 2025Updated 9 months ago
- 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.
- Painless distributed training with torch☆12Apr 28, 2026Updated 2 months ago
- Physics-Enhanced Regression for Initial Value Problems☆20Feb 7, 2024Updated 2 years ago
- Zephyr is a declarative neural network library on top of JAX allowing for easy and fast neural network designing, creation, and manipulat…☆40Mar 21, 2026Updated 3 months ago
- dRiftDM☆15Jun 6, 2026Updated 3 weeks ago
- Package to Train LANs (Likelihood approximation networks)☆16Updated this week
- Scalable drone simulation using JAX.☆134Jun 27, 2026Updated last week
- A Google Chrome Extension and native hosts allowing jspsych to interact with hardware☆16Dec 21, 2018Updated 7 years ago