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 6 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 2 months ago
- Codebase for " Reducing Representation Drift in Online Continual Learning"☆14Jun 8, 2021Updated 5 years ago
- CCLoco: Scaling Up Top-K Error Feedback with Local Optimizers☆27Aug 22, 2025Updated last year
- [Poster; ICLR 2026] [Oral; Neurips OPT2024] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers☆16Apr 15, 2026Updated 5 months ago
- ACCO: An optimization algorithm for sharded distributed LLM training.☆13May 22, 2025Updated last year
- 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.
- [Oral; CVPR'22] Parametric Scattering Networks☆29Nov 10, 2025Updated 10 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
- [WACV'24] Object Re-Identification from Point Clouds☆21Jan 16, 2026Updated 8 months ago
- [ICML 2023] Decentralized SGD and Average-direction SAM are Asymptotically Equivalent☆20Dec 4, 2023Updated 2 years ago
- A comprehensive toolkit for streamlining data editing, search, and inspection for large-scale language model training and interpretabilit…☆21Oct 30, 2025Updated 10 months ago
- ScatNetLight for fast classifications of signals via Scattering Networks☆15Apr 20, 2017Updated 9 years ago
- 📄Small Batch Size Training for Language Models☆87Mar 18, 2026Updated 6 months ago
- 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.
- Code repository for the paper "Geometric Scattering for Graph Data Analysis"☆14Aug 26, 2019Updated 7 years ago
- Training hybrid models for dummies.☆30Nov 1, 2025Updated 10 months ago
- ☆15Sep 29, 2022Updated 3 years ago
- ☆80Oct 19, 2017Updated 8 years ago
- A port of muP to JAX/Haiku☆25Oct 23, 2022Updated 3 years ago
- Thinker project☆16Sep 4, 2024Updated 2 years ago
- An implementation of model parallel autoregressive transformers on GPUs, based on the DeepSpeed library.☆22Nov 28, 2022Updated 3 years ago
- Manuela Demo Apps☆17Mar 28, 2024Updated 2 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
- Deploy on Railway without the complexity - Free Credits Offer • AdConnect your repo and Railway handles the rest with instant previews. Quickly provision container image services, databases, and storage volumes.
- ☆13Apr 7, 2022Updated 4 years ago
- pytorch implementation of SOSELETO☆15Sep 5, 2019Updated 7 years ago
- A data-driven, fast driving simulator for multi-agent coordination under partial observability.☆40Aug 29, 2024Updated 2 years ago
- List and description of the courses I took @ Mila☆15Feb 24, 2019Updated 7 years ago
- Project showing how to develop NKI kernels for Llama 3.2 1B inference☆21May 29, 2025Updated last year
- An implementation of the Equivariant Graph Neural Network (EGNN) layer type for DGL-PyTorch.☆15Dec 27, 2022Updated 3 years ago
- PyTorch implementation of "The Option Keyboard: Combining Skills in Reinforcement Learning" (NeurIPS 2019)☆12Jul 2, 2020Updated 6 years ago
- Implementation of "Few-Shot Single-View 3-D Object Reconstruction with Compositional Priors" ECCV'20 paper.☆14Feb 14, 2021Updated 5 years ago
- Memory Replay with Data Compression (ICLR 2022)☆17Sep 26, 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.
- [NeurIPS 2024 Spotlight] Official implementation for "PACE: Marrying generalization in PArameter-efficient fine-tuning with Consistency r…☆19Jun 5, 2026Updated 3 months ago
- A simple but well-performing "single-hop" visual attention model for the GQA dataset☆20Aug 8, 2019Updated 7 years ago
- Accompanying code for the paper "Beyond Self-Play and Scale: A Behavior Benchmark for Generalization in Autonomous Driving"☆19May 7, 2026Updated 4 months ago
- Implementation for Object Permanence Emerges in a Random Walk along Memory☆22Dec 11, 2022Updated 3 years ago
- Companion code for "Uncertainty-Driven Active Vision for Implicit Scene Reconstruction". This repository contains the code base for a Nex…☆34Oct 13, 2022Updated 3 years ago
- Playground for reinforcement learning algorithms implemented in TensorFlow☆16Oct 18, 2016Updated 9 years ago
- Weight-Averaged Sharpness-Aware Minimization (NeurIPS 2022)☆28Jan 13, 2023Updated 3 years ago