Dion optimizer algorithm
☆438Jan 16, 2026Updated last month
Alternatives and similar repositories for dion
Users that are interested in dion are comparing it to the libraries listed below
Sorting:
- supporting pytorch FSDP for optimizers☆84Dec 8, 2024Updated last year
- Minimal (truly) muP implementation, consistent with TP4 and TP5 papers notation☆14Jan 2, 2026Updated last month
- Muon is an optimizer for hidden layers in neural networks☆2,329Jan 19, 2026Updated last month
- ☆34Sep 10, 2024Updated last year
- Efficient optimizers☆285Dec 20, 2025Updated 2 months ago
- 🧱 Modula software package☆321Aug 18, 2025Updated 6 months ago
- ☆156Updated this week
- Minimalistic large language model 3D-parallelism training☆2,569Feb 19, 2026Updated last week
- NanoGPT (124M) in 2 minutes☆4,679Updated this week
- maximal update parametrization (µP)☆1,686Jul 17, 2024Updated last year
- Muon fsdp 2☆54Aug 8, 2025Updated 6 months ago
- AGaLiTe: Approximate Gated Linear Transformers for Online Reinforcement Learning (Published in TMLR)☆23Oct 15, 2024Updated last year
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆187Jan 19, 2026Updated last month
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆280Nov 24, 2025Updated 3 months ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆222Aug 19, 2024Updated last year
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆190Jan 11, 2026Updated last month
- Code for the paper "Function-Space Learning Rates"☆25Jun 3, 2025Updated 8 months ago
- Helpful tools and examples for working with flex-attention☆1,136Feb 8, 2026Updated 2 weeks ago
- A PyTorch native platform for training generative AI models☆5,098Updated this week
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Apr 17, 2024Updated last year
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,090Aug 26, 2025Updated 6 months ago
- [NeurIPS 2023] and [ICLR 2024] for robustness certification.☆10Nov 30, 2024Updated last year
- Simple & Scalable Pretraining for Neural Architecture Research☆308Dec 6, 2025Updated 2 months ago
- Accelerating MoE with IO and Tile-aware Optimizations☆591Updated this week
- 🚀 Efficient implementations of state-of-the-art linear attention models☆4,428Updated this week
- Muon is Scalable for LLM Training☆1,440Aug 3, 2025Updated 6 months ago
- [Oral; Neurips OPT2024 ] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers☆15Feb 12, 2026Updated 2 weeks ago
- ☆67Mar 21, 2025Updated 11 months ago
- Simple and scalable tools for data-driven pretraining data selection.☆29Jun 9, 2025Updated 8 months ago
- PyTorch Single Controller☆978Updated this week
- Checkpointable dataset utilities for foundation model training☆32Jan 29, 2024Updated 2 years ago
- JAX bindings for Flash Attention v2☆103Feb 19, 2026Updated last week
- https://hf.co/hexgrad/Kokoro-82M☆14Jan 14, 2026Updated last month
- For optimization algorithm research and development.☆558Updated this week
- [ICLR 2026] GRAPE: Group Representational Position Encoding (https://arxiv.org/abs/2512.07805)☆78Jan 27, 2026Updated last month
- Flash-Muon: An Efficient Implementation of Muon Optimizer☆237Jun 15, 2025Updated 8 months ago
- Normalized Transformer (nGPT)☆198Nov 19, 2024Updated last year
- Codebase the paper "The Remarkable Robustness of LLMs: Stages of Inference?"☆19Jun 11, 2025Updated 8 months ago
- Personal solutions to the Triton Puzzles☆20Jul 18, 2024Updated last year