juliusberner / sde_sampler
Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)
☆48Updated 3 weeks ago
Related projects: ⓘ
- ☆90Updated 7 months ago
- ☆30Updated 3 months ago
- PyTorch Implementation of Diffusion Schrodinger Bridge Matching☆109Updated last year
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆15Updated 3 weeks ago
- Implementation of Action Matching☆35Updated last year
- Likelihood Training of Schrödinger Bridge using FBSDEs Theory, ICLR 2022☆71Updated 2 years ago
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆30Updated 5 months ago
- Neural Diffusion Processes☆71Updated last month
- Score-based generative models for compact manifolds☆99Updated 8 months ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆21Updated 8 months ago
- Implementing Controlled Monte Carlo Diffusions (ICLR 2024)☆11Updated 6 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆46Updated 6 months ago
- ☆14Updated 11 months ago
- PyTorch Implementation of DSB for Score Based Generative Modeling. Experiments managed using Hydra.☆138Updated 2 years ago
- Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)☆130Updated 2 years ago
- [ICML2022] Variational Wasserstein gradient flow☆18Updated last year
- PyTorch implementation of Variational Diffusion Models.☆69Updated 4 months ago
- Code for the paper https://arxiv.org/abs/2205.14987v2☆40Updated 5 months ago
- A PyTorch implementation of Bayesian flow networks (Graves et al., 2023).☆21Updated 9 months ago
- Stochastic Normalizing Flows☆72Updated 2 years ago
- Implementation of Action Matching for the Schrödinger equation☆22Updated last year
- code for "Riemannian Flow Matching on General Geometries".☆166Updated 6 months ago
- Entropic Optimal Transport Benchmark (NeurIPS 2023).☆18Updated 5 months ago
- Flow-matching algorithms in JAX☆62Updated last month
- A minimalist implementation of score-based diffusion model☆120Updated 3 years ago
- Transformers with doubly stochastic attention☆40Updated 2 years ago
- Code for GFlowNet-EM, a novel algorithm for fitting latent variable models with compositional latents and an intractable true posterior.☆38Updated 7 months ago
- ☆19Updated 7 months ago
- ☆18Updated this week
- Official implementation of Deep Momentum Schrödinger Bridge☆19Updated 10 months ago