bm371613 / conditional-flow-matching
☆11Updated 5 months ago
Related projects ⓘ
Alternatives and complementary repositories for conditional-flow-matching
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆157Updated last year
- AdaCat☆49Updated 2 years ago
- Easy Hypernetworks in Pytorch and Jax☆96Updated last year
- Flow-matching algorithms in JAX☆74Updated 3 months ago
- ☆33Updated last year
- Official code for "Maximum Likelihood Training of Score-Based Diffusion Models", NeurIPS 2021 (spotlight)☆134Updated 2 years ago
- A JAX implementation of the continuous time formulation of Consistency Models☆83Updated last year
- WIP☆89Updated 3 months ago
- ICML 2023: Reduce, Reuse, Recycle: Composing Energy-Based Diffusion Models with MCMC☆132Updated 3 weeks ago
- Beyond Straight-Through☆90Updated last year
- Visualizations of the theory behind diffusion models.☆74Updated 6 months ago
- ICML 2022: Learning Iterative Reasoning through Energy Minimization☆43Updated last year
- Dataclasses manipulated as numpy arrays (with batching, reshape, slicing,...)☆48Updated last month
- 3D Gaussian Splatting in JAX☆54Updated 5 months ago
- NF-Layers for constructing neural functionals.☆75Updated 10 months ago
- A minimalist implementation of score-based diffusion model☆120Updated 3 years ago
- ☆170Updated 3 weeks ago
- ☆122Updated this week
- The 2D discrete wavelet transform for JAX☆38Updated last year
- Code release for "Stochastic Optimal Control Matching"☆27Updated 2 months ago
- ☆50Updated 10 months ago
- ☆31Updated 2 months ago
- Sequence Modeling with Multiresolution Convolutional Memory (ICML 2023)☆120Updated last year
- Gradient-based constrained optimization for JAX☆26Updated 2 years ago
- Implementation of a multimodal diffusion transformer in Pytorch☆97Updated 4 months ago
- Minimal Implementation of a D3PM in pytorch☆181Updated 6 months ago
- Official implementation of the paper "Unifying GANs and Score-Based Diffusion as Generative Particle Models", NeurIPS 2023☆25Updated 11 months ago
- A practical implementation of GradNorm, Gradient Normalization for Adaptive Loss Balancing, in Pytorch☆77Updated 9 months ago
- Implementation of Perceiver AR, Deepmind's new long-context attention network based on Perceiver architecture, in Pytorch☆86Updated last year
- PyTorch interface for TrueGrad Optimizers☆39Updated last year