GrigoryBartosh / sde_matchingLinks
☆18Updated 6 months ago
Alternatives and similar repositories for sde_matching
Users that are interested in sde_matching are comparing it to the libraries listed below
Sorting:
- Flow-matching algorithms in JAX☆112Updated last year
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆71Updated 9 months ago
- Implicit Convolutional Kernels for Steerable CNNs [NeurIPS'23]☆30Updated 10 months ago
- Diffusion models in PyTorch☆120Updated last week
- ☆198Updated last year
- Implementation of papers in 101 lines of code.☆18Updated 2 years ago
- Collecting research materials on neural samplers with diffusion/flow models☆59Updated 5 months ago
- Clifford-Steerable Convolutional Neural Networks [ICML'24]☆51Updated 7 months ago
- Free-form flows are a generative model training a pair of neural networks via maximum likelihood☆50Updated 6 months ago
- Repo to the paper "Lie Point Symmetry Data Augmentation for Neural PDE Solvers"☆56Updated 2 years ago
- Implementation of Action Matching☆49Updated 2 years ago
- Autoencoders in Function Space: Resolution-invariant autoencoders for functional data based on the functional variational autoencoder (FV…☆31Updated 3 weeks ago
- Official implementation of "Lost in Latent Space: An Empirical Study of Latent Diffusion Models for Physics Emulation"☆50Updated 2 months ago
- diffusionjax is a simple and accessible diffusion models package in JAX☆48Updated 11 months ago
- A parallel ODE solver for PyTorch☆273Updated last year
- Bare-bones implementations of some generative models in Jax: diffusion, normalizing flows, consistency models, flow matching, (beta)-VAEs…☆141Updated 2 years ago
- Official implementation of Stochastic Taylor Derivative Estimator (STDE) NeurIPS2024☆124Updated last year
- Code release for the paper Generative Modeling With Inverse Heat Dissipation☆79Updated 2 years ago
- Implementation of Consistency Models (Song et al 2023) for few-step image generation in Jax.☆19Updated 2 years ago
- Spacetime Geometry of Denoising in Diffusion Models☆21Updated 5 months ago
- ☆36Updated last week
- ☆31Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB). ICLR 2023.☆66Updated last year
- Modern Fixed Point Systems using Pytorch☆125Updated 2 years ago
- Neural Diffusion Processes☆81Updated last year
- Flow Annealed Importance Sampling Bootstrap (FAB) with JAX.☆12Updated last year
- A minimal implementation of Equivariant Neural Fields (https://arxiv.org/abs/2406.05753).☆25Updated last month
- Stochastic Normalizing Flows☆78Updated 4 years ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆35Updated 9 months ago
- PyTorch implementation for our ICLR 2024 paper "Diffusion Generative Flow Samplers: Improving learning signals through partial trajectory…☆26Updated 2 years ago