Information-Fusion-Lab-Umass / NuXLinks
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
Sorting:
- Lightweight library of stochastic gradient MCMC algorithms written in JAX.☆103Updated last year
- Neural likelihood-free methods in PyTorch.☆39Updated 5 years ago
- ☆52Updated 2 years ago
- Probabilistic modeling of tabular data with normalizing flows.☆56Updated last week
- Simulation-based inference benchmark☆97Updated 4 months ago
- Documentation:☆120Updated 2 years ago
- Bayesian algorithm execution (BAX)☆49Updated 3 years ago
- ☆80Updated 3 years ago
- A generic interface for linear algebra backends☆73Updated 3 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.☆45Updated 4 months ago
- We got a stew going!☆27Updated last year
- Normalizing Flows with a resampled base distribution☆47Updated 2 years ago
- Matrix-free linear algebra in JAX.☆119Updated last week
- Riemannian Optimization Using JAX☆49Updated last year
- Code for Gaussian Score Matching Variational Inference☆33Updated 3 months ago
- Differentiable and numerically stable implementation of the matrix exponential☆33Updated 4 years ago
- Stochastic Normalizing Flows☆76Updated 3 years ago
- Conditional density estimation with neural networks☆31Updated 4 months ago
- Stencil computations in JAX☆71Updated last year
- A LinearOperator implementation to wrap the numerical nuts and bolts of GPyTorch☆110Updated 2 months ago
- Riemannian Convex Potential Maps☆67Updated 2 years ago
- Exponential families for JAX☆69Updated this week
- A dead simple Python package for creating custom JAX pytree objects☆63Updated 6 months ago
- Likelihood-free AMortized Posterior Estimation with PyTorch☆128Updated 9 months ago
- A small library for creating and manipulating custom JAX Pytree classes☆56Updated 2 years ago
- Combining smooth constraint for building DAG with normalizing flow in order to replace autoregressive transformations while keeping tract…☆45Updated last year
- Normalizing Flows in Jax☆108Updated 4 years ago
- Implicit Deep Adaptive Design (iDAD): Policy-Based Experimental Design without Likelihoods☆19Updated 3 years ago
- Tools for JAX☆47Updated this week