code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
☆132Jul 24, 2025Updated 7 months ago
Alternatives and similar repositories for adjoint_sampling
Users that are interested in adjoint_sampling are comparing it to the libraries listed below
Sorting:
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆66May 30, 2025Updated 9 months ago
- Code release for "Stochastic Optimal Control Matching"☆40Aug 14, 2024Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆67Mar 11, 2024Updated 2 years ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆47Dec 9, 2024Updated last year
- Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.☆68Apr 11, 2025Updated 11 months ago
- [ICLR 2022] Path integral sampler☆53Aug 21, 2023Updated 2 years ago
- ☆32May 30, 2024Updated last year
- Flow-matching algorithms in JAX☆116Aug 12, 2024Updated last year
- ☆18Sep 27, 2023Updated 2 years ago
- Code for “FlowMM Generating Materials with Riemannian Flow Matching” and "FlowLLM: Flow Matching for Material Generation with Large Langu…☆177Oct 29, 2024Updated last year
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆87Apr 1, 2024Updated last year
- ☆35Sep 12, 2022Updated 3 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆74Mar 14, 2025Updated last year
- ☆12Dec 13, 2023Updated 2 years ago
- ☆13Feb 11, 2025Updated last year
- Collecting research materials on neural samplers with diffusion/flow models☆59Jul 4, 2025Updated 8 months ago
- CUDA implementations of MACE models☆23Aug 19, 2025Updated 7 months ago
- Code for "Accelerating Natural Gradient with Higher-Order Invariance"☆30Jun 28, 2019Updated 6 years ago
- Implementation of Action Matching☆51Jun 5, 2023Updated 2 years ago
- Likelihood Training of Schrödinger Bridge using FBSDEs Theory, ICLR 2022☆88Feb 22, 2022Updated 4 years ago
- AIMNet2: Fast, accurate and transferable neural network interatomic potential☆18Oct 17, 2024Updated last year
- Code for the paper Progressive Inference-Time Annealing of Diffusion Models for Sampling from Boltzmann Densities.☆35Jul 23, 2025Updated 7 months ago
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆52Oct 28, 2022Updated 3 years ago
- Timewarp is a research project using deep learning to accelerate molecular dynamics simulation.☆60Aug 21, 2024Updated last year
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆43Mar 1, 2025Updated last year
- ☆22Feb 4, 2025Updated last year
- Code for paper "Compositional Sculpting of Iterative Generative Processes"☆25Oct 2, 2023Updated 2 years ago
- code for "Riemannian Flow Matching on General Geometries".☆290Mar 13, 2024Updated 2 years ago
- ☆21Dec 1, 2025Updated 3 months ago
- ☆61Apr 10, 2022Updated 3 years ago
- Alchemical machine learning interatomic potentials☆34Nov 8, 2024Updated last year
- Implementation of the Euclidean fast attention (EFA) algorithm☆61Jan 7, 2026Updated 2 months ago
- A repo for transferable sampling of molecular systems☆38Mar 6, 2026Updated 2 weeks ago
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 2 years ago
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆60Sep 26, 2025Updated 5 months ago
- ☆49Jun 17, 2025Updated 9 months ago
- React-OT is a generative transition state search model developed by DeepPrinciple, which uses Optimal Transport (OT) methods to generate …☆41Aug 25, 2025Updated 6 months ago
- TorchCFM: a Conditional Flow Matching library☆2,357Nov 11, 2025Updated 4 months ago
- Equivariant machine learning interatomic potentials in JAX.☆90Feb 10, 2026Updated last month