cloneofsimo / karras-power-ema-tutorialLinks
☆53Updated last year
Alternatives and similar repositories for karras-power-ema-tutorial
Users that are interested in karras-power-ema-tutorial are comparing it to the libraries listed below
Sorting:
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆85Updated last year
- ☆27Updated last year
- ☆23Updated last year
- These papers will provide unique insightful concepts that will broaden your perspective on neural networks and deep learning☆48Updated 2 years ago
- WIP☆93Updated last year
- ☆39Updated last year
- ☆34Updated last year
- ☆22Updated last year
- ☆13Updated last year
- ☆32Updated last year
- JAX implementation ViT-VQGAN☆82Updated 3 years ago
- Minimal multi-gpu implementation of EDM2: "Analyzing and Improving the Training Dynamics of Diffusion Models"☆39Updated last year
- [ICML 2023] Reflected Diffusion Models (https://arxiv.org/abs/2304.04740)☆158Updated 2 years ago
- 🦾 EvalGIM (pronounced as "EvalGym") is an evaluation library for generative image models. It enables easy-to-use, reproducible automatic…☆88Updated 11 months ago
- Train VAE like a boss☆300Updated last year
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆57Updated 8 months ago
- Focused on fast experimentation and simplicity☆75Updated 10 months ago
- A JAX implementation of the continuous time formulation of Consistency Models☆84Updated 2 years ago
- Implementation of a multimodal diffusion transformer in Pytorch☆106Updated last year
- PyTorch implementation for "Parallel Sampling of Diffusion Models", NeurIPS 2023 Spotlight☆148Updated 2 years ago
- Implementation of "Analyzing and Improving the Training Dynamics of Diffusion Models"☆94Updated last year
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Updated last year
- Sparse Autoencoders for Stable Diffusion XL models.☆76Updated 3 weeks ago
- Synthetic Alphabet Dataset☆19Updated 7 months ago
- ☆30Updated last year
- ☆73Updated 2 years ago
- Simple large-scale training of stable diffusion with multi-node support.☆133Updated 2 years ago
- Official codebase for Margin-aware Preference Optimization for Aligning Diffusion Models without Reference (MaPO).☆82Updated last year
- Implementation of Recurrent Interface Network (RIN), for highly efficient generation of images and video without cascading networks, in P…☆204Updated last year
- ☆25Updated last month