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:
- Code for the NLP4Prog workshop paper "Reading StackOverflow Encourages Cheating: Adding Question TextImproves Extractive Code Generation"☆20Aug 10, 2021Updated 4 years ago
- LoRA for arbitrary JAX models and functions☆143Feb 26, 2024Updated 2 years ago
- ☆18Apr 19, 2024Updated 2 years ago
- [Poster; ICLR 2026] [Oral; Neurips OPT2024] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers☆16Apr 15, 2026Updated 3 months ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆94Jan 25, 2024Updated 2 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.
- ☆18Aug 24, 2024Updated last year
- Dreamer on JAX☆16Jan 19, 2022Updated 4 years ago
- A simple implementation of Hamiltonian Monte Carlo in JAX.☆20Feb 8, 2024Updated 2 years ago
- High-performance tokenized language data-loader for Python C++ extension☆15Jul 22, 2024Updated 2 years ago
- Simple TensorFlow implementation of skip-thought vectors☆11Dec 8, 2022Updated 3 years ago
- playing with gpt4☆13Mar 17, 2023Updated 3 years ago
- Guide for fixing 99-100% of cracking sound issues on Dell XPS 15 9570☆11Nov 1, 2018Updated 7 years ago
- ☆15Jun 8, 2026Updated last month
- [ACL'24 Oral] Analysing The Impact of Sequence Composition on Language Model Pre-Training☆24Aug 18, 2024Updated last year
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click. Zero configuration with optimized deployments.
- All notes presented by me and others at a weekly reading group.☆11Aug 26, 2020Updated 5 years ago
- ☆22Dec 8, 2022Updated 3 years ago
- JAX bindings for Flash Attention v2☆108Feb 28, 2026Updated 5 months ago
- nanoGPT using Equinox☆15Mar 3, 2023Updated 3 years ago
- Sharpened Cosine Distance implementation in PyTorch☆10Feb 1, 2022Updated 4 years ago
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- ACCO: An optimization algorithm for sharded distributed LLM training.☆13May 22, 2025Updated last year
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated 2 years ago
- DeltaProduct is a new linear recurrent neural network architecture that uses products of generalized Householder matrices as state-transi…☆15Oct 13, 2025Updated 9 months ago
- Managed Kubernetes at scale on DigitalOcean • AdDigitalOcean Kubernetes includes the control plane, bandwidth allowance, container registry, automatic updates, and more for free.
- Code for the paper "Function-Space Learning Rates"☆23Jun 3, 2025Updated last year
- Subscribe to instant notifications from Student Portal. Don't miss Marvel movie tickets again!☆15Dec 8, 2022Updated 3 years ago
- The accompanying code for "Exploring the limits of decoder-only models trained on public speech recognition corpora" (Ankit Gupta, George…☆21Oct 11, 2024Updated last year
- If it quacks like a tensor...☆59Nov 13, 2024Updated last year
- Very deep VAEs in JAX/Flax☆47Jun 16, 2021Updated 5 years ago
- Deep Learning library for Python. Convnets, recurrent neural networks, and more. Runs on Theano or TensorFlow.☆12Dec 24, 2016Updated 9 years ago
- Actlist Plugin library to development and debugging.☆14Feb 17, 2022Updated 4 years ago
- Server for front-end and editor tools of The Gazelle☆20Apr 17, 2023Updated 3 years ago
- A JAX nn library☆21Sep 9, 2025Updated 10 months ago
- Bare Metal GPUs on DigitalOcean Gradient AI • AdPurpose-built for serious AI teams training foundational models, running large-scale inference, and pushing the boundaries of what's possible.
- TalkingData AdTracking Fraud Detection Challenge☆10May 8, 2018Updated 8 years ago
- Demo codes in our presentation about MXNet in AWS Seoul Summit 2017☆12Apr 24, 2017Updated 9 years ago
- Official implementation of 'A Large-Scale Exploration of mu-Transfer' (CoRR 2024)☆31Jun 5, 2025Updated last year
- [NeurIPS'19] Deep Equilibrium Models Jax Implementation☆43Oct 26, 2020Updated 5 years ago
- Einsum-like high-level array sharding API for JAX☆35Jul 16, 2024Updated 2 years ago
- Relative Positional Encoding for Transformers with Linear Complexity☆65Apr 1, 2022Updated 4 years ago
- ☆11Jan 23, 2017Updated 9 years ago