☆27May 3, 2024Updated last year
Alternatives and similar repositories for min-max-in-dit
Users that are interested in min-max-in-dit are comparing it to the libraries listed below
Sorting:
- ☆24Jun 4, 2024Updated last year
- ☆23Jun 18, 2024Updated last year
- Code for the paper "Function-Space Learning Rates"☆25Jun 3, 2025Updated 9 months ago
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆86Jul 28, 2024Updated last year
- ☆28Oct 7, 2025Updated 4 months ago
- ☆23Oct 15, 2024Updated last year
- Reference implementation of "Softmax Attention with Constant Cost per Token" (Heinsen, 2024)☆24Jun 6, 2024Updated last year
- Display tensors directly from GPU☆11Oct 12, 2025Updated 4 months ago
- Unofficial implementation of paper : Exploring the Space of Key-Value-Query Models with Intention☆12May 24, 2023Updated 2 years ago
- ☆13Jun 3, 2024Updated last year
- recipe for training fully-featured self supervised image jepa models☆12Jun 4, 2025Updated 9 months ago
- Simple (fast) transformer inference in PyTorch with torch.compile + lit-llama code☆10Aug 29, 2023Updated 2 years ago
- Minimal (truly) muP implementation, consistent with TP4 and TP5 papers notation☆14Jan 2, 2026Updated 2 months ago
- Tiny re-implementation of MDM in style of LLaDA and nano-gpt speedrun☆57Mar 10, 2025Updated 11 months ago
- Single-step image generation at 306 FPS. Drifting vs Diffusion head-to-head on CIFAR-10.☆38Feb 13, 2026Updated 2 weeks ago
- [Oral; Neurips OPT2024 ] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers☆15Feb 12, 2026Updated 3 weeks ago
- Minimal implementation of scalable rectified flow transformers, based on SD3's approach☆634Jul 1, 2024Updated last year
- Legible, Scalable, Reproducible Foundation Models with Named Tensors and Jax☆16Jun 16, 2024Updated last year
- ESLTTS dataset☆16Feb 6, 2025Updated last year
- PyTorch implementation of StableMask (ICML'24)☆15Jun 27, 2024Updated last year
- An implementation of PSGD Kron second-order optimizer for PyTorch☆98Jul 24, 2025Updated 7 months ago
- ☆11Sep 13, 2023Updated 2 years ago
- ☆20May 30, 2024Updated last year
- Focused on fast experimentation and simplicity☆80Dec 24, 2024Updated last year
- Fast and memory-efficient exact attention☆29Dec 2, 2024Updated last year
- ☆19Dec 4, 2025Updated 3 months ago
- A library for unit scaling in PyTorch☆133Jul 11, 2025Updated 7 months ago
- ☆48Feb 23, 2025Updated last year
- An AR+AR TTS attempt.☆18Jan 13, 2025Updated last year
- Scalable and Stable Parallelization of Nonlinear RNNS☆29Oct 21, 2025Updated 4 months ago
- ☆20Jun 10, 2024Updated last year
- [ICCV 2025, Highlight] Official Pytorch implementation of the paper: "ReFlex: Text-Guided Editing of Real Images in Rectified Flow via Mi…☆36Aug 1, 2025Updated 7 months ago
- [ICML 2025] Official Implementation of Hierarchical Masked Autoregressive Models with Low-Resolution Token Pivots☆30May 28, 2025Updated 9 months ago
- Griffin MQA + Hawk Linear RNN Hybrid☆89Apr 26, 2024Updated last year
- ☆53Jan 6, 2024Updated 2 years ago
- Supporting code for the blog post on modular manifolds.☆117Sep 26, 2025Updated 5 months ago
- ☆24Sep 25, 2024Updated last year
- Implementation of Diffusion Transformers and Rectified Flow in Jax☆27Jul 9, 2024Updated last year
- Baseline for DCASE 2024 Task 9: "Language-Queried Audio Source Separation"☆26Mar 27, 2024Updated last year