☆72Sep 11, 2026Updated last month
Alternatives and similar repositories for ml-ademamix
Users that are interested in ml-ademamix are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- ☆285Dec 2, 2024Updated last year
- The AdEMAMix Optimizer: Better, Faster, Older.☆187Sep 12, 2024Updated 2 years ago
- Schedule free optimiser implemented in JAX using Optimistix☆14May 29, 2024Updated 2 years ago
- Scalable and Stable Parallelization of Nonlinear RNNS☆33Jun 28, 2026Updated 4 months ago
- Implementation for ACProp ( Momentum centering and asynchronous update for adaptive gradient methdos, NeurIPS 2021)☆17Oct 11, 2021Updated 5 years ago
- Serverless GPU API endpoints on Runpod - Get Bonus Credits • AdSkip the infrastructure headaches. Auto-scaling, pay-as-you-go, no-ops approach lets you focus on innovating your application.
- Combining SOAP and MUON☆24Feb 11, 2025Updated last year
- ☆30Dec 2, 2024Updated last year
- ☆33Sep 10, 2024Updated 2 years ago
- ☆11Mar 14, 2024Updated 2 years ago
- Code for the paper "Function-Space Learning Rates"☆23Jun 3, 2025Updated last year
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆94Oct 30, 2024Updated last year
- A library for unit scaling in PyTorch☆136Jul 11, 2025Updated last year
- Analyzing and Improving Speaker Similarity Assessment in Speech Synthesis☆14Jul 21, 2025Updated last year
- Official repository for paper Routing-Free Mixture-of-Experts.☆19Aug 23, 2026Updated 2 months 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.
- Experiments on the impact of depth in transformers and SSMs.☆48Oct 23, 2025Updated last year
- SLTrain: a sparse plus low-rank approach for parameter and memory efficient pretraining (NeurIPS 2024)☆39Nov 1, 2024Updated last year
- DeMo: Decoupled Momentum Optimization☆202Dec 2, 2024Updated last year
- Toy meta-RL environments for testing algorithms implementations☆10Feb 21, 2024Updated 2 years ago
- Measuring the Signal to Noise Ratio in Language Model Evaluation☆32Aug 19, 2025Updated last year
- Agent-assisted hyperparameter optimization 📈☆16Aug 2, 2026Updated 2 months ago
- ☆23Oct 18, 2026Updated last week
- Integrates Imbue's Cost Aware pareto-Region Bayesian Search (CARBS) with Weights and Biases (WanDB)☆12Mar 17, 2025Updated last year
- ☆10Jun 27, 2024Updated 2 years 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 for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793☆462May 13, 2025Updated last year
- Train a SmolLM-style llm on fineweb-edu in JAX/Flax with an assortment of optimizers.☆19Jul 24, 2025Updated last year
- ☆13Jul 2, 2024Updated 2 years ago
- tutorial for training a BPE tokenizer on multiple GPUs☆13May 5, 2025Updated last year
- Experiments Notebook of "Understanding the Skill Gap in Recurrent Language Models: The Role of the Gather-and-Aggregate Mechanism"☆17Apr 30, 2025Updated last year
- ☆16Jul 16, 2024Updated 2 years ago
- Object-Centric-Representation Library (OCRL): This repo is to explore OCR on various downstream tasks from supervised learning tasks to R…☆12Feb 23, 2024Updated 2 years ago
- solving nethack together! @:@:@:@:@☆26Oct 2, 2026Updated 3 weeks ago
- ☆22Mar 3, 2026Updated 7 months 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.
- Minimal but scalable implementation of large language models in JAX☆34Nov 28, 2025Updated 11 months ago
- ☆13Sep 11, 2026Updated last month
- Parallel Associative Scan for Language Models☆18Jan 8, 2024Updated 2 years ago
- nanoGPT-like codebase for LLM training☆120Nov 7, 2025Updated 11 months ago
- ☆13Jun 3, 2024Updated 2 years ago
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆89Jul 28, 2024Updated 2 years ago
- Einsum-like high-level array sharding API for JAX☆35Jul 16, 2024Updated 2 years ago