A port of muP to JAX/Haiku
☆25Oct 23, 2022Updated 3 years ago
Alternatives and similar repositories for haiku-mup
Users that are interested in haiku-mup are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- JAX notebook showing how to LoRA + GPTQ arbitrary models☆10Aug 8, 2023Updated 2 years ago
- Code for the NLP4Prog workshop paper "Reading StackOverflow Encourages Cheating: Adding Question TextImproves Extractive Code Generation"☆21Aug 10, 2021Updated 4 years ago
- LoRA for arbitrary JAX models and functions☆144Feb 26, 2024Updated 2 years ago
- JAX implementation of GPTQ quantization algorithm☆10Jul 19, 2023Updated 2 years ago
- ☆18Apr 19, 2024Updated 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.
- Unofficial JAX implementation of the SOAP optimizer (https://arxiv.org/abs/2409.11321)☆25Jan 9, 2026Updated 4 months ago
- [Poster; ICLR 2026] [Oral; Neurips OPT2024] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers☆16Apr 15, 2026Updated 3 weeks ago
- ☆15Aug 3, 2021Updated 4 years ago
- ☆18Aug 24, 2024Updated last year
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆20Feb 8, 2024Updated 2 years ago
- My NixOS Configuration.☆20Apr 26, 2026Updated last week
- High-performance tokenized language data-loader for Python C++ extension☆15Jul 22, 2024Updated last year
- playing with gpt4☆14Mar 17, 2023Updated 3 years ago
- ☆13Dec 15, 2025Updated 4 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.
- Guide for fixing 99-100% of cracking sound issues on Dell XPS 15 9570☆11Nov 1, 2018Updated 7 years ago
- JAX bindings for Flash Attention v2☆105Feb 28, 2026Updated 2 months ago
- ☆21Dec 8, 2022Updated 3 years ago
- nanoGPT using Equinox☆15Mar 3, 2023Updated 3 years ago
- Sharpened Cosine Distance implementation in PyTorch☆10Feb 1, 2022Updated 4 years ago
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated last year
- ACCO: An optimization algorithm for sharded distributed LLM training.☆13May 22, 2025Updated 11 months ago
- Exponential families for JAX☆77Apr 30, 2026Updated last week
- All notes presented by me and others at a weekly reading group.☆11Aug 26, 2020Updated 5 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 the paper "Function-Space Learning Rates"☆24Jun 3, 2025Updated 11 months ago
- The accompanying code for "Exploring the limits of decoder-only models trained on public speech recognition corpora" (Ankit Gupta, George…☆20Oct 11, 2024Updated last year
- A package for creating a matrix product state (MPS) from a free fermion (Gaussian) state.☆10Dec 6, 2021Updated 4 years ago
- Very deep VAEs in JAX/Flax☆47Jun 16, 2021Updated 4 years ago
- If it quacks like a tensor...☆59Nov 13, 2024Updated last year
- SOFA Plugin to Reduce Model☆37Jan 7, 2026Updated 4 months ago
- Adaptive Subgoal Search☆20Apr 3, 2023Updated 3 years ago
- A JAX nn library☆21Sep 9, 2025Updated 8 months ago
- Official implementation of 'A Large-Scale Exploration of mu-Transfer'☆32Jun 5, 2025Updated 11 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.
- 11th Solution of Kaggle TalkingData AdTracking Fraud Detection Challenge☆10May 10, 2018Updated 7 years ago
- Identifying the language of input text using character-level n-grams, with support for 45 languages☆11Dec 26, 2022Updated 3 years ago
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆43Oct 26, 2020Updated 5 years ago
- Auction algorithm for solving linear assignment problem (LAP)☆75Jun 21, 2022Updated 3 years ago
- Python 3.6 and TensorFlow implementation of the AReS and MaRS algorithms☆11Jun 23, 2019Updated 6 years ago
- diffGrad: An Optimization Method for Convolutional Neural Networks☆55Oct 12, 2022Updated 3 years ago
- ☆11Jan 23, 2017Updated 9 years ago