microsoft / dionLinks
Dion optimizer algorithm
☆403Updated this week
Alternatives and similar repositories for dion
Users that are interested in dion are comparing it to the libraries listed below
Sorting:
- 🧱 Modula software package☆315Updated 3 months ago
- Simple & Scalable Pretraining for Neural Architecture Research☆304Updated last week
- Load compute kernels from the Hub☆348Updated this week
- FlexAttention based, minimal vllm-style inference engine for fast Gemma 2 inference.☆320Updated last month
- ☆285Updated last year
- ☆68Updated last year
- supporting pytorch FSDP for optimizers☆84Updated last year
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆179Updated 5 months ago
- Home for "How To Scale Your Model", a short blog-style textbook about scaling LLMs on TPUs☆724Updated 2 weeks ago
- Fault tolerance for PyTorch (HSDP, LocalSGD, DiLoCo, Streaming DiLoCo)☆455Updated last week
- CIFAR-10 speedruns: 94% in 2.6 seconds and 96% in 27 seconds☆330Updated 3 weeks ago
- Efficient optimizers☆275Updated last month
- Minimal yet performant LLM examples in pure JAX☆207Updated last week
- Supporting code for the blog post on modular manifolds.☆104Updated 2 months ago
- ☆224Updated 2 weeks ago
- PyTorch-native post-training at scale☆559Updated last week
- Normalized Transformer (nGPT)☆193Updated last year
- An implementation of PSGD Kron second-order optimizer for PyTorch☆97Updated 4 months ago
- ☆106Updated this week
- Scalable and Performant Data Loading☆352Updated this week
- For optimization algorithm research and development.☆548Updated 3 weeks ago
- ☆933Updated last month
- ☆91Updated last year
- Quantized LLM training in pure CUDA/C++.☆221Updated this week
- ☆288Updated this week
- A library for unit scaling in PyTorch☆132Updated 5 months ago
- Implementation of Diffusion Transformer (DiT) in JAX☆298Updated last year
- Training API and CLI☆253Updated this week
- Open-source framework for the research and development of foundation models.☆648Updated last week
- seqax = sequence modeling + JAX☆168Updated 4 months ago