Information-Fusion-Lab-Umass / NuX
Normalizing Flows using JAX
☆82Updated 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.☆99Updated last year
- Neural likelihood-free methods in PyTorch.☆39Updated 4 years ago
- ☆49Updated last year
- ☆129Updated 3 weeks ago
- A generic interface for linear algebra backends☆71Updated 6 months ago
- Simulation-based inference benchmark☆95Updated last month
- Probabilistic modeling of tabular data with normalizing flows.☆55Updated 7 months ago
- Exponential families for JAX☆60Updated last week
- ☆80Updated 3 years ago
- Riemannian Optimization Using JAX☆47Updated last year
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆121Updated 4 months ago
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆43Updated this week
- Normalizing Flows with a resampled base distribution☆44Updated 2 years ago
- Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods☆17Updated 3 years ago
- Multiple dispatch over abstract array types in JAX.☆113Updated 2 weeks ago
- Sliced Iterative Generator (SIG) & Gaussianizing Iterative Slicing (GIS)☆36Updated last year
- Approximate inference for Markov Gaussian processes using iterated Kalman smoothing, in JAX☆98Updated last year
- Riemannian Convex Potential Maps☆67Updated last year
- Code for efficiently sampling functions from GP(flow) posteriors☆68Updated 4 years ago
- Documentation:☆119Updated last year
- Stochastic Normalizing Flows☆73Updated 3 years ago
- Matrix-free linear algebra in JAX.☆113Updated last week
- Sequential Neural Likelihood☆38Updated 5 years ago
- Library for normalizing flows and neural flows.☆24Updated 2 years ago
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆17Updated 11 months ago
- Conditional density estimation with neural networks☆29Updated 4 months ago
- A framework for composing Neural Processes in Julia☆76Updated 3 years ago