Kei18 / tiny-tiny-diffusion
minimal diffusion model for self-study
☆19Updated last year
Alternatives and similar repositories for tiny-tiny-diffusion:
Users that are interested in tiny-tiny-diffusion are comparing it to the libraries listed below
- Jax/Flax implementation of Denoising Diffusion Implicit Models☆17Updated 2 years ago
- Flow-matching algorithms in JAX☆82Updated 5 months ago
- PyTorch implementations of normalizing flow and its variants.☆74Updated 3 years ago
- Code release for "Stochastic Optimal Control Matching"☆30Updated 5 months ago
- ☆44Updated 8 months ago
- JAX implementation of the JKOnet* architecture presented in "Learning Diffusion at Lightspeed".☆26Updated 3 weeks ago
- A demo shows how to combine Langevin dynamics with score matching for generative models.☆36Updated 4 years ago
- Codes for the paper "A mathematical perspective on Transformers".☆34Updated 6 months ago
- [ICLR 2022] Path integral sampler☆43Updated last year
- CycleQD is a framework for parameter space model merging.☆27Updated last month
- ☆18Updated last year
- Light and Optimal Schrödinger Bridge Matching (ICML 2024) official PyTorch implementation☆34Updated 5 months ago
- ☆33Updated last year
- Transformers with doubly stochastic attention☆44Updated 2 years ago
- Pytorch-like dataloaders in JAX.☆67Updated 3 months ago
- Gradient-based constrained optimization for JAX☆28Updated 2 years ago
- Open source code for EigenGame.☆29Updated last year
- FID computation in Jax/Flax.☆26Updated 6 months ago
- flexible meta-learning in jax☆12Updated last year
- 拡散モデルを学びたい初学者向けです。書籍「コンピュータビジョン最前線 Summer 2023」の「イマドキノ拡散モデル」の解説をベースに、CIFER-10で画像生成をします☆16Updated last year
- Neural Diffusion Processes☆76Updated 5 months ago
- A minimal Pytorch Implementation of Stochastically Quantized Variational AutoEncoder (SQ-VAE) by Sony☆30Updated last year
- [Neurips 2021]Diffusion Normalizing Flow (DiffFlow)☆117Updated last year
- ☆31Updated last month
- Improved sampling via learned diffusions (ICLR2024) and an optimal control perspective on diffusion-based generative modeling (TMLR2024)☆54Updated 4 months ago
- Annotated Flow Matching paper☆157Updated 4 months ago
- Implementation of PSGD optimizer in JAX☆26Updated 2 weeks ago
- A Small diffusion model in PyTorch.☆16Updated 9 months ago
- Simple and extensible hypergradient for PyTorch☆16Updated last year
- A repo where I play with conditional flow approaches for learning time-varying vector-fields.☆11Updated 7 months ago