code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
☆130Jul 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:
- Code release for "Stochastic Optimal Control Matching"☆39Aug 14, 2024Updated last year
- Reward fine-tuning for Stable Diffusion models based on stochastic optimal control, including Adjoint Matching☆64May 30, 2025Updated 9 months ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆67Mar 11, 2024Updated last year
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆46Dec 9, 2024Updated last year
- Code for “FlowMM Generating Materials with Riemannian Flow Matching” and "FlowLLM: Flow Matching for Material Generation with Large Langu…☆175Oct 29, 2024Updated last year
- ☆32May 30, 2024Updated last year
- ☆34Sep 12, 2022Updated 3 years ago
- Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.☆68Apr 11, 2025Updated 10 months ago
- [ICLR 2022] Path integral sampler☆52Aug 21, 2023Updated 2 years ago
- Flow-matching algorithms in JAX☆115Aug 12, 2024Updated last year
- CUDA implementations of MACE models☆23Aug 19, 2025Updated 6 months ago
- ☆13Feb 11, 2025Updated last year
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆87Apr 1, 2024Updated last year
- ☆12Dec 13, 2023Updated 2 years ago
- Implementation of Action Matching☆51Jun 5, 2023Updated 2 years ago
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 2 years ago
- AIMNet2: Fast, accurate and transferable neural network interatomic potential☆18Oct 17, 2024Updated last year
- Code for "Accelerating Natural Gradient with Higher-Order Invariance"☆30Jun 28, 2019Updated 6 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆72Mar 14, 2025Updated 11 months ago
- ☆48Jun 17, 2025Updated 8 months ago
- A repo for transferable sampling of molecular systems☆36Updated this week
- Collection of tutorials to use the MACE machine learning force field.☆53Jan 22, 2026Updated last month
- [NeurIPS'25 AI4Mat] Nequix: Training a foundation model for materials on a budget and [arXiv'26] Phonon fine-tuning (PFT)☆68Updated this week
- Alchemical machine learning interatomic potentials☆34Nov 8, 2024Updated last year
- Likelihood Training of Schrödinger Bridge using FBSDEs Theory, ICLR 2022☆88Feb 22, 2022Updated 4 years ago
- [NeurIPS 2024] Official implementation of the Efficiently Scaled Attention Interatomic Potential☆59Sep 26, 2025Updated 5 months ago
- Implementation of the Euclidean fast attention (EFA) algorithm☆61Jan 7, 2026Updated last month
- 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.☆59Aug 21, 2024Updated last year
- Collecting research materials on neural samplers with diffusion/flow models☆59Jul 4, 2025Updated 7 months ago
- Zero Shot Molecular Generation via Similarity Kernels☆28Aug 27, 2025Updated 6 months ago
- MACE_Osaka24 models☆25Dec 19, 2024Updated last year
- Parallel waveform generation with DiffusionGAN☆17Mar 26, 2022Updated 3 years ago
- Equivariant machine learning interatomic potentials in JAX.☆87Feb 10, 2026Updated 2 weeks ago
- code for "Riemannian Flow Matching on General Geometries".☆289Mar 13, 2024Updated last year
- DIfferentiable MOlecular Simulator (DIMOS): A pyTorch based framework for machine-learning enhanced simulations☆29Nov 11, 2025Updated 3 months ago
- Official pytorch implementation for our ICLR 2023 paper "Latent State Marginalization as a Low-cost Approach for Improving Exploration".☆24Feb 9, 2023Updated 3 years ago
- Code repository for Trajectory Flow Matching☆101Oct 31, 2024Updated last year
- ☆22Feb 4, 2025Updated last year