Bayesian inference with Python and Jax.
☆34Nov 23, 2022Updated 3 years ago
Alternatives and similar repositories for jax-bayes
Users that are interested in jax-bayes are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Neat Bayesian machine learning examples☆58Feb 10, 2026Updated 3 months ago
- Optimisation on Diffeomorphisms☆12Feb 17, 2025Updated last year
- Benchmark for learning stiff problems using physics-informed machine learning☆13Dec 15, 2021Updated 4 years ago
- A fast tensor library for c++.☆11Sep 23, 2015Updated 10 years ago
- Simple (and cheap!) neural network uncertainty estimation☆82Oct 7, 2025Updated 7 months ago
- 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.
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆25Oct 2, 2023Updated 2 years ago
- Interactive textbook on state space models☆12Apr 11, 2022Updated 4 years ago
- Official PyTorch implementation of OTiS: An open model for general time series analysis.☆18Apr 1, 2026Updated last month
- Supplementary Material to accompany the paper, DJ Warne, SA Sisson, C Drovandi (2019) Acceleration of expensive computations in Bayesian…☆13Oct 23, 2020Updated 5 years ago
- ☆18Apr 2, 2023Updated 3 years ago
- Code for utilising VAE as means of doing exact MCMC inference in complex high-dimensional space☆14Jun 20, 2023Updated 2 years ago
- Bayesian Bandits☆71Sep 3, 2023Updated 2 years ago
- Code to reproduce the experiments in The Mirage of Action-Dependent Baselines in Reinforcement Learning.☆17Aug 2, 2018Updated 7 years ago
- Train text generation model with JavaScript.☆15Jul 14, 2024Updated 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.
- Tutorial notebooks for SciFM24☆11Apr 2, 2024Updated 2 years ago
- Probabilistic deep learning using JAX☆15Feb 8, 2025Updated last year
- ☆14Aug 22, 2024Updated last year
- This repository presents a series of analysis on the performance of Physics-Informed Neural Networks in vibrational systems. The limitati…☆13Apr 15, 2023Updated 3 years ago
- ☆17Aug 6, 2024Updated last year
- Noah's master thesis on Quantum Bayesian Neural Networks☆13Aug 2, 2021Updated 4 years ago
- ☆14Sep 11, 2022Updated 3 years ago
- Bayesian optimization of discrete sequences☆21Mar 27, 2022Updated 4 years ago
- Official implementation of GPX: Gaussian Process Regression with Interpretable Sample-wise Feature Weights (published on TNNLS)☆17Feb 8, 2022Updated 4 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- Code for "Deep Convolutional Networks as shallow Gaussian Processes"☆16Jun 21, 2019Updated 6 years ago
- MCHMC: sampler from an arbitrary differentiable distribution☆76Dec 9, 2025Updated 5 months ago
- Improving predictions of Bayesian neural nets via local linearization, AISTATS 2021☆15Dec 30, 2022Updated 3 years ago
- On Uncertainty, Tempering, and Data Augmentation in Bayesian Classification☆21Apr 1, 2022Updated 4 years ago
- Perceiver (transformer variant) implemented in JAX and Flax☆13Mar 29, 2021Updated 5 years ago
- orbital MCMC☆10Jun 17, 2021Updated 4 years ago
- Fast Kolmogorov-Arnold Network in JAX, initial experiments☆16May 20, 2024Updated 2 years ago
- ☆15Sep 8, 2025Updated 8 months ago
- Code of PriorVAE: encoding spatial priors with variational autoencoders☆12Jul 13, 2023Updated 2 years 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.
- Idiomatic access to Starlink applications from Python.☆16Jul 19, 2018Updated 7 years ago
- ☆16Mar 5, 2026Updated 2 months ago
- Cross Atlas Remapping via Optimal Transport☆12Dec 14, 2023Updated 2 years ago
- Modular pipeline for design assistance. PYPi Package: https://pypi.org/project/vai-lab☆14Dec 22, 2023Updated 2 years ago
- Simple audio AE☆13Nov 10, 2024Updated last year
- Course for ISP: craft of data visualization☆22Sep 1, 2022Updated 3 years ago
- ☆16Jul 7, 2025Updated 10 months ago