harryjo97 / riemannian-diffusion-mixture-torchLinks
PyTorch implementation for "Generative Modeling on Manifolds Through Mixture of Riemannian Diffusion Processes" (ICML 2024).
☆13Updated last year
Alternatives and similar repositories for riemannian-diffusion-mixture-torch
Users that are interested in riemannian-diffusion-mixture-torch are comparing it to the libraries listed below
Sorting:
- code for "Riemannian Flow Matching on General Geometries".☆283Updated last year
- [ICML 2025] The official implementation of the paper "On the Guidance of Flow Matching"☆79Updated 4 months ago
- Score-based generative models for compact manifolds☆119Updated 2 years ago
- Official codebase for the paper "How to build a consistency model: Learning flow maps via self-distillation" (NeurIPS 2025).☆70Updated 3 months ago
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆84Updated last year
- Spacetime Geometry of Denoising in Diffusion Models☆21Updated 6 months ago
- ☆160Updated last year
- [NeurIPS 2025] Official repository for "Energy Matching: Unifying Flow Matching and Energy-Based Models for Generative Modeling"☆198Updated 3 months ago
- The official PyTorch implementation of Manifold Preserving Guided Diffusion (MPGD)☆38Updated last year
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆62Updated 8 months ago
- STeP: a general and scalable framework for solving video inverse problems with spatiotemporal diffusion priors☆29Updated 7 months ago
- Learning Large-scale Neural Fields via Context Pruned Meta-Learning (NeurIPS 2023)☆27Updated 2 years ago
- ☆26Updated 2 years ago
- ☆55Updated last year
- Code for the paper: Rotating Features for Object Discovery☆53Updated last year
- Official implementation of Bidirectional Diffusion Bridge Models☆23Updated 8 months ago
- [CVPR 2025 Oral] Improving Diffusion Inverse Problem Solving with Decoupled Noise Annealing☆174Updated last month
- NeurIPS23 "Flow Factorized Representation Learning"☆41Updated last month
- Official PyTorch implementation of the paper: Flow Matching in Latent Space☆331Updated last year
- Code repository for Trajectory Flow Matching☆99Updated last year
- Meta Flow Matching: Integrating Vector Fields on the Wasserstein Manifold☆69Updated 10 months ago
- A Tutorial for Diffusion Models☆60Updated 2 years ago
- Annotated Flow Matching paper☆227Updated last year
- ☆205Updated last year
- Official Code Repository for the paper "Generative Modeling on Manifolds Through Mixture of Riemannian Diffusion Processes" (ICML 2024).☆15Updated last year
- InverseBench (ICLR 2025 spotlight)☆94Updated 3 months ago
- Code for PnP-Flow: Plug-and-Play image restoration with Flow Matching (ICLR 2025)☆140Updated 4 months ago
- [NeurIPS 2024] Official implementation of NeurIPS 2024 paepr "Flow Priors for Linear Inverse Problems via Iterative Corrupted Trajectory …☆25Updated 11 months ago
- ☆67Updated 6 months ago
- [NeurIPS'23 Spotlight] Learning Probabilistic Symmetrization for Architecture Agnostic Equivariance (LPS), in PyTorch☆30Updated last year