juliusberner / sde_sampler
Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
☆61Updated last week
Alternatives and similar repositories for sde_sampler:
Users that are interested in sde_sampler are comparing it to the libraries listed below
- ☆138Updated last year
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆30Updated 3 months ago
- Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.☆47Updated last month
- ☆32Updated 9 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆55Updated last year
- Code release for "Stochastic Optimal Control Matching"☆30Updated 7 months ago
- Implementing Controlled Monte Carlo Diffusions (ICLR 2024)☆15Updated 5 months ago
- Score-based generative models for compact manifolds☆108Updated last year
- Implementation of Action Matching☆39Updated last year
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- Flow-matching algorithms in JAX☆86Updated 7 months ago
- Likelihood Training of Schrödinger Bridge using FBSDEs Theory, ICLR 2022☆81Updated 3 years ago
- ☆16Updated last year
- Official implementation of Deep Momentum Schrödinger Bridge☆24Updated last year
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆62Updated 11 months ago
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆147Updated last month
- Stochastic Normalizing Flows☆76Updated 3 years ago
- Code repository for Particle Denoising Diffusion Sampler☆12Updated 11 months ago
- A PyTorch implementation of Bayesian flow networks (Graves et al., 2023).☆25Updated last year
- Neural Diffusion Processes☆76Updated 7 months ago
- Code for the paper https://arxiv.org/abs/2205.14987v2☆48Updated 11 months ago
- ☆26Updated last year
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆23Updated last year
- ☆19Updated last month
- Official Jax Implementation of MD4 Masked Diffusion Models☆67Updated 3 weeks ago
- Transformers with doubly stochastic attention☆45Updated 2 years ago
- ☆24Updated last week
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆47Updated 2 years ago
- ☆25Updated 7 months ago