A simple reference implementation of the single-worker MuLoCo optimizer in Jax & PyTorch. MuLoCo-1 has been shown to outperfrom Muon and have larger critical batch sizes.
☆32Feb 27, 2026Updated 4 months ago
Alternatives and similar repositories for muloco-1
Users that are interested in muloco-1 are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- An efficient implementation of learned optimizers in PyTorch☆62Jun 23, 2026Updated 3 weeks ago
- CCLoco: Scaling Up Top-K Error Feedback with Local Optimizers☆26Aug 22, 2025Updated 10 months ago
- [Poster; ICLR 2026] [Oral; Neurips OPT2024] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers☆16Apr 15, 2026Updated 3 months ago
- ACCO: An optimization algorithm for sharded distributed LLM training.☆13May 22, 2025Updated last year
- ☆32Aug 28, 2020Updated 5 years ago
- Managed Database hosting by DigitalOcean • AdPostgreSQL, MySQL, MongoDB, Kafka, Valkey, and OpenSearch available. Automatically scale up storage and focus on building your apps.
- [Oral; CVPR'22] Parametric Scattering Networks☆29Nov 10, 2025Updated 8 months ago
- ISMIR 2021: Curriculum Learning for Imbalanced Classification in Large Vocabulary Automatic Chord Recognition☆10Nov 8, 2021Updated 4 years ago
- An alternative EQA paradigm and informative benchmark + models (BMVC 2019, ViGIL 2019 spotlight)☆25Jun 22, 2022Updated 4 years ago
- ☆20Oct 21, 2022Updated 3 years ago
- This is the official implementation of the ICML 2023 paper - Can Forward Gradient Match Backpropagation ?☆13May 31, 2023Updated 3 years ago
- [ICML 2023] Decentralized SGD and Average-direction SAM are Asymptotically Equivalent☆20Dec 4, 2023Updated 2 years ago
- ScatNetLight for fast classifications of signals via Scattering Networks☆15Apr 20, 2017Updated 9 years ago
- ☆20Apr 11, 2024Updated 2 years ago
- 📄Small Batch Size Training for Language Models☆82Mar 18, 2026Updated 4 months ago
- GPU virtual machines on DigitalOcean Gradient AI • AdGet to production fast with high-performance AMD and NVIDIA GPUs you can spin up in seconds. The definition of operational simplicity.
- Code repository for the paper "Geometric Scattering for Graph Data Analysis"☆14Aug 26, 2019Updated 6 years ago
- Training hybrid models for dummies.☆31Nov 1, 2025Updated 8 months ago
- ☆11Dec 8, 2022Updated 3 years ago
- "Towards Scaling Difference Target Propagation by Learning Backprop Targets" (ICML 2022)☆13Jan 17, 2023Updated 3 years ago
- A Python Library for Probabilistic Sparse Coding with Non-Standard Priors and Superpositions☆20Oct 17, 2019Updated 6 years ago
- [TMLR 2024] Revisiting Random Weight Perturbation for Efficiently Improving Generalization☆12Oct 18, 2024Updated last year
- A port of muP to JAX/Haiku☆25Oct 23, 2022Updated 3 years ago
- An implementation of model parallel autoregressive transformers on GPUs, based on the DeepSpeed library.☆21Nov 28, 2022Updated 3 years ago
- Official repository for ICCV 2023: Get the Best of Both Worlds: Improving Accuracy and Transferability by Grassmann Class☆13Oct 16, 2023Updated 2 years ago
- Wordpress hosting with auto-scaling - Free Trial Offer • AdFully Managed hosting for WordPress and WooCommerce businesses that need reliable, auto-scalable performance. Cloudways SafeUpdates now available.
- ☆88Jun 16, 2025Updated last year
- ☆13Apr 7, 2022Updated 4 years ago
- pytorch implementation of SOSELETO☆15Sep 5, 2019Updated 6 years ago
- Official Repository for "Communication Efficient Federated Learning with Generalized Heavy-Ball Momentum", accepted at TMLR 2025☆28Jul 14, 2025Updated last year
- ☆19Apr 16, 2025Updated last year
- An implementation of the Equivariant Graph Neural Network (EGNN) layer type for DGL-PyTorch.☆15Dec 27, 2022Updated 3 years ago
- [NeurIPS 2024 Spotlight] Official implementation for "PACE: Marrying generalization in PArameter-efficient fine-tuning with Consistency r…☆19Jun 5, 2026Updated last month
- This repository includes code to reproduce the tables in "Loss Landscapes are All You Need: Neural Network Generalization Can Be Explaine…☆39Mar 2, 2023Updated 3 years ago
- ☆19Apr 16, 2022Updated 4 years ago
- Managed hosting for WordPress and PHP on Cloudways • AdManaged hosting for WordPress, Magento, Laravel, or PHP apps, on multiple cloud providers. Deploy in minutes on Cloudways by DigitalOcean.
- A simple but well-performing "single-hop" visual attention model for the GQA dataset☆20Aug 8, 2019Updated 6 years ago
- ☆17Dec 11, 2022Updated 3 years ago
- ScatWave is a Torch implementation of scattering using CUDA☆21Jan 18, 2018Updated 8 years ago
- Comparison of bandit algorithms from the Reinforcement Learning bible.☆17Jun 6, 2018Updated 8 years ago
- Implementation for Object Permanence Emerges in a Random Walk along Memory☆22Dec 11, 2022Updated 3 years ago
- [TPAMI 2023] Low Dimensional Landscape Hypothesis is True: DNNs can be Trained in Tiny Subspaces☆42Jun 29, 2022Updated 4 years ago
- ☆17May 1, 2023Updated 3 years ago