Dion optimizer algorithm
☆483Jun 7, 2026Updated this week
Alternatives and similar repositories for dion
Users that are interested in dion are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- supporting pytorch FSDP for optimizers☆84Dec 8, 2024Updated last year
- ☆34Sep 10, 2024Updated last year
- [NeurIPS 2023] and [ICLR 2024] for robustness certification.☆10Nov 30, 2024Updated last year
- Muon is an optimizer for hidden layers in neural networks☆2,642May 24, 2026Updated 2 weeks ago
- 🧱 Modula software package☆333Aug 18, 2025Updated 9 months ago
- Simple, predictable pricing with DigitalOcean hosting • AdAlways know what you'll pay with monthly caps and flat pricing. Enterprise-grade infrastructure trusted by 600k+ customers.
- Minimal (truly) muP implementation, consistent with TP4 and TP5 papers notation☆14Jan 2, 2026Updated 5 months ago
- Efficient optimizers☆330May 13, 2026Updated 3 weeks ago
- Muon fsdp 2☆59Aug 8, 2025Updated 10 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆194Jan 19, 2026Updated 4 months ago
- ☆194Jun 3, 2026Updated last week
- 🚀 Efficiently (pre)training foundation models with native PyTorch features, including FSDP for training and SDPA implementation of Flash…☆286Nov 24, 2025Updated 6 months ago
- Helpful tools and examples for working with flex-attention☆1,195May 28, 2026Updated last week
- NanoGPT (124M) in 90 seconds☆5,337Jun 2, 2026Updated last week
- Code for the paper "Function-Space Learning Rates"☆24Jun 3, 2025Updated last year
- Deploy to Railway using AI coding agents - Free Credits Offer • AdUse Claude Code, Codex, OpenCode, and more. Autonomous software development now has the infrastructure to match with Railway.
- maximal update parametrization (µP)☆1,722Jul 17, 2024Updated last year
- Minimalistic large language model 3D-parallelism training☆2,711May 26, 2026Updated 2 weeks ago
- Simple & Scalable Pretraining for Neural Architecture Research☆333Mar 31, 2026Updated 2 months ago
- ☆13Mar 10, 2026Updated 2 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆198May 30, 2026Updated last week
- Minimal (400 LOC) implementation Maximum (multi-node, FSDP) GPT training☆132Apr 17, 2024Updated 2 years ago
- A PyTorch native platform for training generative AI models☆5,416Updated this week
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆379Nov 15, 2025Updated 6 months ago
- ☆10Oct 27, 2023Updated 2 years ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Muon is Scalable for LLM Training☆1,487Aug 3, 2025Updated 10 months ago
- Minimalistic 4D-parallelism distributed training framework for education purpose☆2,216Aug 26, 2025Updated 9 months ago
- Code for Spectral Norm of Convolutional Layers with Circular and Zero Paddings and Efficient Bound of Lipschitz Constant for Convolutiona…☆15Feb 2, 2024Updated 2 years ago
- Official repository for DistFlashAttn: Distributed Memory-efficient Attention for Long-context LLMs Training☆223Aug 19, 2024Updated last year
- Ship correct and fast LLM kernels to PyTorch☆150Jan 14, 2026Updated 4 months ago
- [Poster; ICLR 2026] [Oral; Neurips OPT2024] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers☆16Apr 15, 2026Updated last month
- 🚀 Efficient implementations for emerging model architectures☆5,182Updated this week
- ☆13Jul 2, 2024Updated last year
- For optimization algorithm research and development.☆570May 6, 2026Updated last month
- End-to-end encrypted cloud storage - Proton Drive • AdSpecial offer: 40% Off Yearly / 80% Off First Month. Protect your most important files, photos, and documents from prying eyes.
- ☆69Mar 21, 2025Updated last year
- Accelerating MoE with IO and Tile-aware Optimizations☆707May 14, 2026Updated 3 weeks ago
- PyTorch Single Controller☆1,042Updated this week
- Spectral Sphere Optimizer☆118Mar 23, 2026Updated 2 months ago
- JAX bindings for Flash Attention v2☆106Feb 28, 2026Updated 3 months ago
- 🔥 A minimal training framework for scaling FLA models☆391Apr 22, 2026Updated last month
- Simple and scalable tools for data-driven pretraining data selection.☆29Jun 9, 2025Updated last year