Implementation of Action Matching
☆51Jun 5, 2023Updated 2 years ago
Alternatives and similar repositories for jam
Users that are interested in jam are comparing it to the libraries listed below
Sorting:
- ☆32May 30, 2024Updated last year
- Implementation of Action Matching for the Schrödinger equation☆25Jun 18, 2023Updated 2 years ago
- orbital MCMC☆10Jun 17, 2021Updated 4 years ago
- Code release for "Stochastic Optimal Control Matching"☆39Aug 14, 2024Updated last year
- ☆22Feb 4, 2025Updated last year
- Reading Group @mila-iqia on Computational Optimal Transport for Machine Learning Applications☆13Jun 3, 2022Updated 3 years ago
- Proximal Optimal Transport Modeling of Population Dynamics (AISTATS 2022)☆22Jun 19, 2023Updated 2 years ago
- Lagrangian formulation of Doob's h-transform allowing for efficient rare event sampling☆57Mar 26, 2025Updated 11 months ago
- ☆18Mar 16, 2022Updated 3 years ago
- Multiscale earth mover's distance methods for tree and manifold structured data☆12Mar 8, 2025Updated 11 months ago
- [ICML'25] The Price of Freedom: Exploring Expressivity and Runtime Tradeoffs in Equivariant Tensor Products☆15Jul 16, 2025Updated 7 months ago
- code for "Adjoint Sampling: Highly Scalable Diffusion Samplers via Adjoint Matching"☆130Jul 24, 2025Updated 7 months ago
- ☆52Feb 27, 2023Updated 3 years ago
- Taxonomy of Benchmarks in Graph Representation Learning☆21Dec 1, 2022Updated 3 years ago
- The Superposition of Diffusion Models Using the Itô Density Estimator☆52Mar 20, 2025Updated 11 months ago
- code for "Generalized Schrödinger Bridge Matching" (ICLR 2024).☆87Apr 1, 2024Updated last year
- Deep Generalized Schrödinger Bridge, NeurIPS 2022 Oral☆52Oct 28, 2022Updated 3 years ago
- ☆40Jan 27, 2026Updated last month
- ☆34Sep 12, 2022Updated 3 years ago
- ☆35Updated this week
- ☆39Jun 26, 2024Updated last year
- Graph Learning with JAX☆14Jul 11, 2022Updated 3 years ago
- Transformer Based Embeddings of Wasserstein Distance☆22Feb 18, 2026Updated 2 weeks ago
- [ICML 2025] Repurposing pre-trained score-based generative models for transition path sampling by minimizing the Onsager-Machlup (OM) act…☆26Jan 14, 2026Updated last month
- An implementation of the differentially private variational inference algorithm for NumPyro.☆16Sep 3, 2024Updated last year
- MESS: Modern Electronic Structure Simulations☆43Sep 26, 2025Updated 5 months ago
- Code repository for Particle Denoising Diffusion Sampler☆15Apr 2, 2024Updated last year
- DiffSyn: A Generative Diffusion Approach to Materials Synthesis Planning (Nature Computational Science, 2026)☆28Feb 10, 2026Updated 3 weeks ago
- A selection of neural network models ported from torchvision for JAX & Flax.☆45Jul 19, 2025Updated 7 months ago
- A repo for transferable sampling of molecular systems☆36Updated this week
- ☆18Jun 26, 2023Updated 2 years ago
- ☆80Sep 4, 2021Updated 4 years ago
- Very deep VAEs in JAX/Flax☆46Jun 16, 2021Updated 4 years ago
- Uncover meaningful structures of latent spaces learned by generative models with flows!☆43May 10, 2024Updated last year
- Code repository of the paper "BooVAE: Boosting Approach for Continual Learning of VAE" published at NeurIPS 2021. https://arxiv.org/abs/1…☆13Oct 18, 2021Updated 4 years ago
- An unofficial implementation of the Fourier Neural Operator in Flax☆19May 25, 2024Updated last year
- Bayesian and Maximum Likelihood Implementation of the Normalizing Flow Network (NFN): https://arxiv.org/abs/1907.08982☆22Dec 10, 2020Updated 5 years ago
- [NeurIPS 2025] Implementation of "SceneSplat++: A Large Dataset and Comprehensive Benchmark for Language Gaussian Splatting"☆45May 16, 2025Updated 9 months ago
- code for "Riemannian Flow Matching on General Geometries".☆290Mar 13, 2024Updated last year