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
Sorting:
- ☆33Nov 4, 2024Updated 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☆285Dec 20, 2025Updated 2 months ago
- ☆23Jun 18, 2024Updated last year
- ☆48Feb 23, 2025Updated last year
- WIP☆94Aug 13, 2024Updated last year
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆57Mar 10, 2025Updated 11 months ago
- ☆34Sep 10, 2024Updated last year
- Code for the paper "Stack Attention: Improving the Ability of Transformers to Model Hierarchical Patterns"☆18Mar 15, 2024Updated last year
- ☆28Oct 7, 2025Updated 4 months ago
- ☆23Oct 15, 2024Updated last year
- PyTorch interface for TrueGrad Optimizers☆43Aug 8, 2023Updated 2 years ago
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆634Jul 1, 2024Updated last year
- ☆39Apr 27, 2024Updated last year
- ☆27May 3, 2024Updated last year
- [NeurIPS 2024] Official PyTorch Implementation of "FlowTurbo: Towards Real-time Flow-Based Image Generation with Velocity Refiner"☆73Oct 21, 2025Updated 4 months ago
- Load & manage evolving datasets efficiently☆23Aug 22, 2025Updated 6 months ago
- Visualising Losses in Deep Neural Networks☆16Jul 17, 2024Updated last year
- Synthetic Alphabet Dataset☆19Mar 27, 2025Updated 11 months ago
- ☆18Aug 24, 2024Updated last year
- ☆252Dec 2, 2024Updated last year
- Writing FLUX in Triton☆42Sep 22, 2024Updated last year
- The 2D discrete wavelet transform for JAX☆45Feb 28, 2023Updated 3 years ago
- Train VAE like a boss☆313Oct 21, 2024Updated last year
- [Preprint] GMem: A Modular Approach for Ultra-Efficient Generative Models☆43Mar 11, 2025Updated 11 months ago
- Utilities for PyTorch distributed☆25Feb 27, 2025Updated last year
- Implementation of Diffusion Transformers and Rectified Flow in Jax☆27Jul 9, 2024Updated last year
- High-performance tokenized language data-loader for Python C++ extension☆14Jul 22, 2024Updated last year
- Unofficial implementation for Sigmoid Loss for Language Image Pre-Training☆11Sep 26, 2023Updated 2 years ago
- Code for PuzzleJAX, a benchmark for reasoning and learning, that reimplements PuzzleScript, a concise and expressive DSL and game engine …☆25Updated this week
- ☆29Jul 9, 2024Updated last year
- ☆24Jun 4, 2024Updated last year
- DiCE: The Infinitely Differentiable Monte-Carlo Estimator☆32Jul 28, 2023Updated 2 years ago
- Learning Motion from Low-Rank Adaptation☆46Jun 15, 2024Updated last year
- Autoregressive Image Generation☆31Jun 13, 2025Updated 8 months ago
- Flash Attention in 300-500 lines of CUDA/C++☆36Aug 22, 2025Updated 6 months ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Jun 5, 2025Updated 8 months ago
- ☆13Jun 3, 2024Updated last year