Information-Fusion-Lab-Umass / NuX
Normalizing Flows using JAX
☆83Updated last year
Alternatives and similar repositories for NuX:
Users that are interested in NuX are comparing it to the libraries listed below
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆102Updated last year
- ☆52Updated 2 years ago
- Neural likelihood-free methods in PyTorch.☆39Updated 5 years ago
- Probabilistic modeling of tabular data with normalizing flows.☆56Updated last week
- A generic interface for linear algebra backends☆73Updated last month
- Likelihood-free AMortized Posterior Estimation with PyTorch☆126Updated 7 months ago
- Code for Gaussian Score Matching Variational Inference☆33Updated last month
- ☆80Updated 3 years ago
- Normalizing Flows with a resampled base distribution☆46Updated 2 years ago
- ☆154Updated last month
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Stochastic Normalizing Flows☆76Updated 3 years ago
- Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods☆19Updated 3 years ago
- Matrix-free linear algebra in JAX.☆116Updated 3 months ago
- Simulation-based inference benchmark☆96Updated 2 months ago
- Multiple dispatch over abstract array types in JAX.☆115Updated this week
- Documentation:☆119Updated last year
- Sliced Iterative Generator (SIG) & Gaussianizing Iterative Slicing (GIS)☆36Updated 2 years ago
- Normalizing Flows in Jax☆107Updated 4 years ago
- JAX Arrays for human consumption☆92Updated last year
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Sequential Neural Likelihood☆41Updated 5 years ago
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆18Updated last year
- Conditional density estimation with neural networks☆30Updated 2 months ago
- A dead simple Python package for creating custom JAX pytree objects☆63Updated 4 months ago
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆98Updated last year
- Riemannian Optimization Using JAX☆48Updated last year
- Library for Bayesian Neural Networks in PyTorch (first version as published in ProbProg2020)☆42Updated 3 years ago
- Code for efficiently sampling functions from GP(flow) posteriors☆72Updated 4 years ago