Focused on fast experimentation and simplicity
☆80Dec 24, 2024Updated last year
Alternatives and similar repositories for diffusion-speedrun
Users that are interested in diffusion-speedrun are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆33Nov 4, 2024Updated last year
- Lightweight package that tracks and summarizes code changes using LLMs (Large Language Models)☆34Feb 27, 2025Updated last year
- supporting pytorch FSDP for optimizers☆84Dec 8, 2024Updated last year
- research impl of Native Sparse Attention (2502.11089)☆63Feb 19, 2025Updated last year
- Efficient optimizers☆294Updated this week
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆58Mar 10, 2025Updated last year
- ☆23Jun 18, 2024Updated last year
- ☆48Feb 23, 2025Updated last year
- WIP☆94Aug 13, 2024Updated last year
- ☆28Oct 7, 2025Updated 5 months ago
- PyTorch interface for TrueGrad Optimizers☆43Aug 8, 2023Updated 2 years ago
- ☆34Sep 10, 2024Updated last year
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆635Jul 1, 2024Updated last year
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆32Jul 28, 2023Updated 2 years ago
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Mar 15, 2024Updated 2 years ago
- ☆18Aug 24, 2024Updated last year
- Load & manage evolving datasets efficiently☆23Aug 22, 2025Updated 7 months ago
- ☆256Dec 2, 2024Updated last year
- ☆27Jul 9, 2024Updated last year
- ☆39Apr 27, 2024Updated last year
- ☆27May 3, 2024Updated last year
- ☆23Oct 15, 2024Updated last year
- Train VAE like a boss☆313Oct 21, 2024Updated last year
- Learning Motion from Low-Rank Adaptation☆46Jun 15, 2024Updated last year
- A locally trained model of Stoney Nakoda has been developed and released. You can access the working model here or train your own instanc…☆10Oct 29, 2025Updated 4 months ago
- Maximal Update Parametrization (μP) with Flax & Optax.☆16Dec 27, 2023Updated 2 years ago
- Visualising Losses in Deep Neural Networks☆16Jul 17, 2024Updated last year
- HomebrewNLP in JAX flavour for maintable TPU-Training☆51Jan 20, 2024Updated 2 years ago
- Official repo of dataset-decomposition paper [NeurIPS 2024]☆21Jan 8, 2025Updated last year
- ☆33Jan 6, 2025Updated last year
- Synthetic Alphabet Dataset☆19Mar 27, 2025Updated 11 months ago
- [NeurIPS 2024] Official PyTorch Implementation of "FlowTurbo: Towards Real-time Flow-Based Image Generation with Velocity Refiner"☆74Oct 21, 2025Updated 5 months ago
- A collection of hash and RNG functions that rely on hardware AES☆24Jan 4, 2025Updated last year
- Automatically take good care of your preemptible TPUs☆37May 15, 2023Updated 2 years ago
- ☆37Dec 25, 2025Updated 2 months ago
- Autoregressive Image Generation☆31Jun 13, 2025Updated 9 months ago
- Writing FLUX in Triton☆42Sep 22, 2024Updated last year
- Implementation of PSGD optimizer in JAX☆35Dec 31, 2024Updated last year
- Tiny AutoEncoder for Stable Diffusion Videos☆36Oct 5, 2024Updated last year