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
Sorting:
- Benchmark for learning stiff problems using physics-informed machine learning☆12Dec 15, 2021Updated 4 years ago
- Neat Bayesian machine learning examples☆58Feb 10, 2026Updated last month
- A fast tensor library for c++.☆11Sep 23, 2015Updated 10 years ago
- Code to reproduce the experiments in The Mirage of Action-Dependent Baselines in Reinforcement Learning.☆17Aug 2, 2018Updated 7 years ago
- GPT implementation in Flax☆18Jan 8, 2022Updated 4 years ago
- ☆18Apr 2, 2023Updated 2 years ago
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆25Oct 2, 2023Updated 2 years ago
- Cross Atlas Remapping via Optimal Transport☆12Dec 14, 2023Updated 2 years ago
- Bayesian Bandits☆70Sep 3, 2023Updated 2 years ago
- Sparsity-promoting Kernel Dynamic Mode Decomposition for Nonlinear Dynamical Systems☆31Jul 30, 2022Updated 3 years ago
- Simple (and cheap!) neural network uncertainty estimation☆79Oct 7, 2025Updated 5 months ago
- MCHMC: sampler from an arbitrary differentiable distribution☆75Dec 9, 2025Updated 3 months ago
- Supporting code for the paper "Dangers of Bayesian Model Averaging under Covariate Shift"☆33Oct 19, 2022Updated 3 years ago
- ☆11Mar 24, 2021Updated 4 years ago
- ☆10Mar 29, 2023Updated 2 years ago
- JAX port of Persistent Independent Particles☆37Feb 4, 2023Updated 3 years ago
- Reference implementation of Thin and Deep Gaussian Processes (NeurIPS 2023)☆14Nov 25, 2024Updated last year
- An implementation of a neural network training routine using derivative information in Pytorch.☆10Dec 19, 2020Updated 5 years ago
- Geographic Data Science in Python - UFMG'19☆12Mar 26, 2019Updated 6 years ago
- ☆14Jan 23, 2025Updated last year
- Code for utilising VAE as means of doing exact MCMC inference in complex high-dimensional space☆14Jun 20, 2023Updated 2 years ago
- ☆16Jul 7, 2025Updated 8 months ago
- Publication of Python code used to train ModalPINN☆11May 5, 2022Updated 3 years ago
- ConfigParser class with AES-256 symmetric encryption support☆10Mar 8, 2024Updated 2 years ago
- ESA's anomaly dataset of real satellite telemetry from three ESA missions☆16Jun 5, 2025Updated 9 months ago
- Python library to convert EEG datasets to a BIDS compatible dataset☆17Jan 21, 2025Updated last year
- diffusionjax is a simple and accessible diffusion models package in JAX☆50Jan 19, 2025Updated last year
- ☆13Oct 16, 2024Updated last year
- Implementation of the ICLR 2022 paper "Phase Collapse in Neural Networks."☆10Mar 21, 2022Updated 3 years ago
- Code that accompanies the paper Guided Deep Kernel Learning☆10Jul 2, 2023Updated 2 years ago
- Gym implementation of connector to Deepmind lab☆12Mar 26, 2019Updated 6 years ago
- Multi-Objective Causal Bayesian Optimisation, a new paradigm for finding Pareto-optimal interventions in multi-outcome causal models☆16Jun 2, 2025Updated 9 months ago
- A few examples for LMAX disruptor☆17Aug 1, 2011Updated 14 years ago
- Augmented Sliced Wasserstein Distances☆10Jun 14, 2025Updated 8 months ago
- A framework for creating databases for surrogate models of complex physics codes.☆12Updated this week
- A trading demo application☆16Aug 28, 2013Updated 12 years ago
- ☆17Mar 4, 2025Updated last year
- sEEG software☆17Feb 11, 2026Updated 3 weeks ago
- A data-driven framework for control of nonlinear flows with Koopman Model Predictive Control☆10May 21, 2018Updated 7 years ago