jaxns

Probabilistic Framework

A probabilistic programming framework for Bayesian inference and model comparison using nested sampling and JAX.

Probabilistic Programming and Nested sampling in JAX

GitHub

147 stars
5 watching
10 forks
Language: Python
last commit: 8 days ago
Linked from 1 awesome list

bayesian-computingbayesian-methodsjaxmodel-comparisonnested-samplingprobabilistic-programmingscientific-computingscientific-machine-learning

Backlinks from these awesome lists:

Related projects:

Repository Description Stars
blackjax-devs/blackjax A Bayesian inference library designed to provide modular and customizable samplers for probabilistic programming 846
camdavidsonpilon/probabilistic-programming-and-bayesian-methods-for-hackers An introduction to Bayesian methods and probabilistic programming for software developers 26,798
joashc/csharp-probability-monad A monadic probabilistic programming framework for Bayesian modelling and inference in C#. 96
mcleonard/sampyl An MCMC sampler library for Bayesian estimation using Python 330
alonfnt/bayex An implementation of Bayesian optimization using Gaussian process regression and JAX for efficient numerical computations 84
jaxgaussianprocesses/gpjax Provides a low-level interface to Gaussian process models in JAX for flexible extension and customisation 461
rlouf/mcx Tools and methods for Bayesian deep learning using probabilistic programming. 325
google/edward2 A tool for writing probabilistic models and manipulating their computation 679
doobwa/mcmc.jl MCMC algorithms implemented in Julia for Bayesian modeling 35
robinthibaut/skbel A Python framework for Bayesian inference and regression using Gaussian processes. 22
ctallec/pyvarinf A Python package facilitating Bayesian Deep Learning methods with Variational Inference for PyTorch 359
maxsklar/bayespy A Python library for Bayesian inference and multinomial mixture modeling 108
wlwkgus/noisynaturalgradient PyTorch implementation of Noisy Natural Gradient as Variational Inference for Bayesian Neural Networks 121
hakaru-dev/hakaru A language for specifying and transforming probabilistic models and inference algorithms in Haskell. 311
aaltoml/kalman-jax This project develops an approximate inference framework for temporal Gaussian processes using Kalman filtering and smoothing methods. 95