lollcat / fab-jaxLinks
Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.
☆11Updated 11 months ago
Alternatives and similar repositories for fab-jax
Users that are interested in fab-jax are comparing it to the libraries listed below
Sorting:
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- ☆52Updated 2 years ago
- Code for Gaussian Score Matching Variational Inference☆33Updated 3 months ago
- Stainless neural networks in JAX☆34Updated 2 months ago
- ☆26Updated last year
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Conditional Flow Matching using JAX☆13Updated last year
- Visualize, create, and operate on pytrees in the most intuitive way possible.☆45Updated 4 months ago
- A collection of graph neural networks implementations in JAX☆32Updated last year
- Implementation of papers in 101 lines of code.☆18Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆58Updated last year
- Code for the paper "Towards Reliable Simulation-Based Inference with Balanced Neural Ratio Estimation".☆14Updated 2 years ago
- Minimal JAX implementation of k-nearest neighbors using a k-d tree.☆37Updated this week
- Normalizing Flows using JAX☆83Updated last year
- Gaussian Mixture Models in Jax☆19Updated 4 months ago
- Simulation-based inference in JAX☆31Updated last month
- GATSBI: Generative Adversarial Training for Simulation-Based Inference☆18Updated last year
- Things that make me feel productive☆15Updated 2 years ago
- Stochastic Normalizing Flows☆76Updated 3 years ago
- Bind any function written in another language to JAX with support for JVP/VJP/batching/jit compilation☆72Updated last month
- Collecting research materials on neural samplers with diffusion/flow models☆48Updated last week
- Library for normalizing flows and neural flows.☆24Updated 3 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆24Updated 2 years ago
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆19Updated last year
- ☆13Updated last year
- Probabilistic modeling of tabular data with normalizing flows.☆56Updated last week
- Neural likelihood-free methods in PyTorch.☆39Updated 5 years ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆34Updated 5 months ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆38Updated 2 years ago
- Implementation of Action Matching☆44Updated 2 years ago