cloneofsimo / karras-power-ema-tutorial
☆51Updated 10 months ago
Related projects ⓘ
Alternatives and complementary repositories for karras-power-ema-tutorial
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆68Updated 3 months ago
- These papers will provide unique insightful concepts that will broaden your perspective on neural networks and deep learning☆46Updated last year
- ☆13Updated 5 months ago
- ☆31Updated 2 months ago
- ☆28Updated 3 weeks ago
- ☆26Updated 6 months ago
- ☆21Updated 5 months ago
- JAX implementation ViT-VQGAN☆77Updated 2 years ago
- Official codebase for Margin-aware Preference Optimization for Aligning Diffusion Models without Reference (MaPO).☆61Updated 5 months ago
- WIP☆89Updated 3 months ago
- Official implementation of "Is This Loss Informative? Faster Text-to-Image Customization by Tracking Objective Dynamics" (NeurIPS 2023)☆37Updated last year
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆89Updated 9 months ago
- ☆71Updated last year
- ☆33Updated 6 months ago
- ☆31Updated 11 months ago
- A JAX implementation of the continuous time formulation of Consistency Models☆83Updated last year
- ☆34Updated 3 weeks ago
- ☆24Updated 5 months ago
- My take on Flow Matching☆25Updated 2 weeks ago
- The codebase of our paper "Improving the Training of Rectified Flows"☆83Updated last month
- Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"☆26Updated 8 months ago
- A big_vision inspired repo that implements a generic Auto-Encoder class capable in representation learning and generative modeling.☆30Updated 4 months ago
- ☆26Updated 2 months ago
- Official implementation of the paper The Hidden Language of Diffusion Models☆69Updated 10 months ago
- ☆78Updated 10 months ago
- ☆73Updated 4 months ago
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated last year
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight☆125Updated last year
- ☆14Updated last month
- Official code for "On Calibrating Diffusion Probabilistic Models"☆29Updated last year