dimitra-maoutsa / odes_for_sdes
Deterministic particle dynamics for simulating Fokker-Planck probability flows
☆24Updated last year
Related projects ⓘ
Alternatives and complementary repositories for odes_for_sdes
- Hamiltonian Dynamics with Non-Newtonian Momentum for Rapid Sampling☆36Updated 3 years ago
- Implementation of Action Matching for the Schrödinger equation☆22Updated last year
- ☆49Updated last year
- Repository for Deterministic Particle Flow Control framework☆10Updated 2 years ago
- Implementation of Action Matching☆36Updated last year
- code for "Neural Conservation Laws A Divergence-Free Perspective".☆35Updated last year
- Implementation of various equivariant models in JAX☆12Updated 7 months ago
- Refining continuous-in-depth neural networks☆39Updated 3 years ago
- Implementation of approximate free-energy minimization in PyTorch☆18Updated 3 years ago
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆51Updated 8 months ago
- Code for "'Hey, that's not an ODE:' Faster ODE Adjoints via Seminorms" (ICML 2021)☆86Updated 2 years ago
- Forward mode laplacian implemented in JAX tracer☆29Updated 2 weeks ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆48Updated last year
- Supplementary code for the paper "Stationary Kernels and Gaussian Processes on Lie Groups and their Homogeneous Spaces"☆41Updated last year
- Library for normalizing flows and neural flows.☆24Updated 2 years ago
- ☆19Updated last month
- Stochastic Normalizing Flows☆73Updated 2 years ago
- Riemannian Optimization Using JAX☆45Updated last year
- Source code for Large-Scale Wasserstein Gradient Flows (NeurIPS 2021)☆29Updated 2 years ago
- Combining smooth constraint for building DAG with normalizing flow in order to replace autoregressive transformations while keeping tract…☆44Updated last year
- Code for Gaussian Score Matching Variational Inference☆28Updated last month
- ☆15Updated 3 years ago
- Riemannian Convex Potential Maps☆68Updated last year
- Normalizing Flows with a resampled base distribution☆44Updated 2 years ago
- Matrix-free linear algebra in JAX.☆106Updated 2 months ago
- Continuous-time gradient flow for generative modeling and variational inference☆29Updated 6 years ago
- Euclidean Wasserstein-2 optimal transportation☆44Updated last year
- Normalizing Flows using JAX☆82Updated 11 months ago
- A collection of graph neural networks implementations in JAX☆31Updated 11 months ago
- [ICML 2024] Official implementation for "Beyond ELBOs: A Large-Scale Evaluation of Variational Methods for Sampling".☆25Updated 2 months ago