facebookresearch / adjoint_sampling
code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"
☆47Updated this week
Alternatives and similar repositories for adjoint_sampling
Users that are interested in adjoint_sampling are comparing it to the libraries listed below
Sorting:
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling☆49Updated last month
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆57Updated last year
- Implementation of various equivariant models in JAX☆12Updated last year
- Annotated implementations of equivariant (graph) neural networks in Jax: EGNN, SEGNN, NequIP.☆33Updated 2 months ago
- SE(3) Equivariant Augmented Coupling Flows. NeurIPS 2023.☆24Updated last year
- Tools for building equivariant polynomials on reductive Lie groups.☆32Updated last year
- Implementation of Action Matching for the Schrödinger equation☆24Updated last year
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆32Updated 5 months ago
- ☆11Updated last year
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Implementation of Action Matching☆41Updated last year
- ☆19Updated 3 months ago
- ☆28Updated 2 years ago
- Code for the paper Iterated Denoising Energy Matching for Sampling from Boltzmann Densities.☆53Updated last month
- Erwin: A Tree-based Hierarchical Transformer for Large-scale Physical Systems [ICML'25]☆51Updated last week
- An implementation of ESM2 in Equinox+JAX☆25Updated 3 months ago
- Official code repository of paper Equivariance via Minimal Frame Averaging for More Symmetries and Efficiency.☆17Updated 3 months ago
- Steerable E(3) GNN in jax☆24Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆11Updated 11 months ago
- ☆52Updated 2 years ago
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆61Updated 2 months ago
- ☆28Updated last month
- Deterministic particle dynamics for simulating Fokker-Planck probability flows☆24Updated 2 years ago
- This Denoising Force Field (DFF) codebase provides a Pytorch framework for the method presented in Two for one: Diffusion models and forc…☆60Updated 11 months ago
- E(n) Equivariant GNN in jax☆11Updated last year
- [ICLR 2022] Path integral sampler☆46Updated last year
- PySCF on IPU☆42Updated last year
- Building Blocks for Equivariant Neural Networks in e3nn and PyTorch 2.0☆12Updated last month
- Code for the paper https://arxiv.org/abs/2402.04997☆77Updated last year
- ☆33Updated last month