For optimization algorithm research and development.
☆558Updated this week
Alternatives and similar repositories for optimizers
Users that are interested in optimizers are comparing it to the libraries listed below
Sorting:
- ☆252Dec 2, 2024Updated last year
- Efficient optimizers☆285Dec 20, 2025Updated 2 months ago
- Schedule-Free Optimization in PyTorch☆2,256May 21, 2025Updated 9 months ago
- TensorDict is a pytorch dedicated tensor container.☆1,009Updated this week
- [Oral; Neurips OPT2024 ] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers☆15Feb 12, 2026Updated 2 weeks ago
- Official codebase used to develop Vision Transformer, SigLIP, MLP-Mixer, LiT and more.☆3,368May 19, 2025Updated 9 months ago
- A PyTorch native platform for training generative AI models☆5,098Updated this week
- Helpful tools and examples for working with flex-attention☆1,136Feb 8, 2026Updated 2 weeks ago
- PyTorch native quantization and sparsity for training and inference☆2,696Feb 22, 2026Updated last week
- MLCommons Algorithmic Efficiency is a benchmark and competition measuring neural network training speedups due to algorithmic improvement…☆413Feb 11, 2026Updated 2 weeks ago
- Annotated version of the Mamba paper☆497Feb 27, 2024Updated 2 years ago
- torch-optimizer -- collection of optimizers for Pytorch☆3,163Mar 22, 2024Updated last year
- PyTorch extensions for high performance and large scale training.☆3,400Apr 26, 2025Updated 10 months ago
- ☆316Jun 21, 2024Updated last year
- maximal update parametrization (µP)☆1,686Jul 17, 2024Updated last year
- Muon is an optimizer for hidden layers in neural networks☆2,329Jan 19, 2026Updated last month
- A PyTorch repo for data loading and utilities to be shared by the PyTorch domain libraries.☆1,248Updated this week
- Minimalistic large language model 3D-parallelism training☆2,569Feb 19, 2026Updated last week
- supporting pytorch FSDP for optimizers☆84Dec 8, 2024Updated last year
- functorch is JAX-like composable function transforms for PyTorch.☆1,436Aug 21, 2025Updated 6 months ago
- ☆34Sep 10, 2024Updated last year
- Pipeline Parallelism for PyTorch☆785Aug 21, 2024Updated last year
- Named tensors with first-class dimensions for PyTorch☆332Jun 14, 2023Updated 2 years ago
- ☆22Nov 9, 2024Updated last year
- ☆31Jan 23, 2026Updated last month
- Python-based research interface for blackbox and hyperparameter optimization, based on the internal Google Vizier Service.☆1,629Feb 17, 2026Updated last week
- TorchMultimodal is a PyTorch library for training state-of-the-art multimodal multi-task models at scale.☆1,698Updated this week
- Pytorch implementation of preconditioned stochastic gradient descent (Kron and affine preconditioner, low-rank approximation precondition…☆190Jan 11, 2026Updated last month
- Ring attention implementation with flash attention☆986Sep 10, 2025Updated 5 months ago
- Recipes are a standard, well supported set of blueprints for machine learning engineers to rapidly train models using the latest research…☆340Updated this week
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆91Oct 30, 2024Updated last year
- Code for exploring Based models from "Simple linear attention language models balance the recall-throughput tradeoff"☆248Jun 6, 2025Updated 8 months ago
- PyTorch native post-training library☆5,689Updated this week
- NanoGPT (124M) in 2 minutes☆4,679Updated this week
- A library for unit scaling in PyTorch☆133Jul 11, 2025Updated 7 months ago
- Official Implementation of "ADOPT: Modified Adam Can Converge with Any β2 with the Optimal Rate"☆434Dec 12, 2024Updated last year
- Type annotations and dynamic checking for a tensor's shape, dtype, names, etc.☆1,472May 2, 2025Updated 9 months ago
- Hackable and optimized Transformers building blocks, supporting a composable construction.☆10,353Feb 20, 2026Updated last week
- Foundation Architecture for (M)LLMs☆3,135Apr 11, 2024Updated last year