lollcat / fab-jaxLinks
Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.
☆12Updated last year
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☆25Updated 2 years ago
- ☆52Updated 2 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆66Updated last year
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 4 years ago
- Code for Gaussian Score Matching Variational Inference☆35Updated 10 months ago
- Implementation of papers in 101 lines of code.☆18Updated 2 years ago
- ☆30Updated 2 years ago
- Stochastic Normalizing Flows☆78Updated 4 years ago
- Stainless neural networks in JAX☆34Updated last month
- Normalizing Flows with a resampled base distribution☆47Updated 3 years ago
- Implementation of Action Matching☆49Updated 2 years ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆41Updated last year
- ☆10Updated 2 years ago
- Library for normalizing flows and neural flows.☆26Updated 3 years ago
- A Metropolis-Hastings MCMC sampler accelerated via diffusion models☆16Updated last year
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆40Updated 3 years ago
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆41Updated 3 years ago
- A collection of graph neural networks implementations in JAX☆35Updated 2 years ago
- Riemannian Optimization Using JAX☆53Updated 2 years ago
- Normalizing Flows using JAX☆86Updated 2 years ago
- ☆18Updated 3 years ago
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆26Updated 2 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆71Updated 9 months ago
- Conditional Flow Matching using JAX☆13Updated 2 years ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆50Updated 6 months ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆26Updated 2 years ago
- Collecting research materials on neural samplers with diffusion/flow models☆59Updated 6 months ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆56Updated 2 years ago
- ☆31Updated last year
- Einstein Fields official implementation in JAX.☆67Updated 4 months ago