For optimization algorithm research and development.
☆561Mar 3, 2026Updated 2 weeks ago
Alternatives and similar repositories for optimizers
Users that are interested in optimizers are comparing it to the libraries listed below
Sorting:
- ☆255Dec 2, 2024Updated last year
- Efficient optimizers☆294Updated this week
- Schedule-Free Optimization in PyTorch☆2,265May 21, 2025Updated 9 months ago
- [Oral; Neurips OPT2024 ] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers☆16Feb 12, 2026Updated last month
- TensorDict is a pytorch dedicated tensor container.☆1,015Updated this week
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆416Updated this week
- ☆22Nov 9, 2024Updated last year
- A PyTorch native platform for training generative AI models☆5,162Updated this week
- torch-optimizer -- collection of optimizers for Pytorch☆3,167Mar 22, 2024Updated last year
- [ICML 2024] SIRFShampoo: Structured inverse- and root-free Shampoo in PyTorch (https://arxiv.org/abs/2402.03496)☆15Nov 4, 2024Updated last year
- Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.☆3,380May 19, 2025Updated 10 months ago
- ☆317Jun 21, 2024Updated last year
- PyTorch native quantization and sparsity for training and inference☆2,730Mar 14, 2026Updated last week
- Annotated version of the Mamba paper☆499Feb 27, 2024Updated 2 years ago
- Muon is an optimizer for hidden layers in neural networks☆2,398Jan 19, 2026Updated 2 months ago
- Helpful tools and examples for working with flex-attention☆1,157Feb 8, 2026Updated last month
- supporting pytorch FSDP for optimizers☆84Dec 8, 2024Updated last year
- PyTorch extensions for high performance and large scale training.☆3,403Apr 26, 2025Updated 10 months ago
- maximal update parametrization (µP)☆1,690Jul 17, 2024Updated last year
- ☆34Sep 10, 2024Updated last year
- Minimalistic large language model 3D-parallelism training☆2,617Feb 19, 2026Updated last month
- Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…☆343Mar 9, 2026Updated last week
- Named tensors with first-class dimensions for PyTorch☆332Jun 14, 2023Updated 2 years ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆435Dec 12, 2024Updated last year
- A library for unit scaling in PyTorch☆133Jul 11, 2025Updated 8 months ago
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆192Jan 11, 2026Updated 2 months ago
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,633Feb 17, 2026Updated last month
- NanoGPT (124M) in 2 minutes☆4,848Updated this week
- functorch is JAX-like composable function transforms for PyTorch.☆1,437Aug 21, 2025Updated 7 months ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆92Oct 30, 2024Updated last year
- PyTorch native post-training library☆5,703Mar 14, 2026Updated last week
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,250Mar 14, 2026Updated last week
- ☆29Sep 30, 2025Updated 5 months ago
- Pipeline Parallelism for PyTorch☆785Aug 21, 2024Updated last year
- Meta Lingua: a lean, efficient, and easy-to-hack codebase to research LLMs.☆4,757Jul 18, 2025Updated 8 months ago
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆249Jun 6, 2025Updated 9 months ago
- The simplest, fastest repository for training/finetuning medium-sized GPTs.☆190Jan 19, 2026Updated 2 months ago
- FFCV: Fast Forward Computer Vision (and other ML workloads!)☆2,986Jun 16, 2024Updated last year
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,373Updated this week