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☆145Feb 26, 2024Updated 2 years ago
- JAX implementation of GPTQ quantization algorithm☆10Jul 19, 2023Updated 2 years ago
- ☆18Apr 19, 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 and start building anything your business needs.
- Public Release of Plan2vec Implementation in pyTorch☆57Oct 28, 2022Updated 3 years ago
- Unofficial but Efficient Implementation of "Mamba: Linear-Time Sequence Modeling with Selective State Spaces" in JAX☆93Jan 25, 2024Updated 2 years ago
- [Oral; Neurips OPT2024 ] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers☆16Feb 12, 2026Updated last month
- ☆15Aug 3, 2021Updated 4 years ago
- ☆18Aug 24, 2024Updated last year
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- 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☆14Jul 22, 2024Updated last year
- playing with gpt4☆14Mar 17, 2023Updated 3 years ago
- Virtual machines for every use case on DigitalOcean • AdGet dependable uptime with 99.99% SLA, simple security tools, and predictable monthly pricing with DigitalOcean's virtual machines, called Droplets.
- ☆13Dec 15, 2025Updated 3 months ago
- Guide for fixing 99-100% of cracking sound issues on Dell XPS 15 9570☆11Nov 1, 2018Updated 7 years ago
- [ACL'24 Oral] Analysing The Impact of Sequence Composition on Language Model Pre-Training☆23Aug 18, 2024Updated last year
- nanoGPT using Equinox☆15Mar 3, 2023Updated 3 years ago
- ISMIR 2021: Curriculum Learning for Imbalanced Classification in Large Vocabulary Automatic Chord Recognition☆10Nov 8, 2021Updated 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 10 months ago
- stochastic bfloat16 based optimizer library☆21Dec 4, 2024Updated last year
- Code for the paper "Function-Space Learning Rates"☆25Jun 3, 2025Updated 9 months ago
- Proton VPN Special Offer - Get 70% off • AdSpecial partner offer. Trusted by over 100 million users worldwide. Tested, Approved and Recommended by Experts.
- Implementations of the renormalization group-based diffusion model (RGDM).☆16Mar 10, 2025Updated last year
- A package for creating a matrix product state (MPS) from a free fermion (Gaussian) state.☆10Dec 6, 2021Updated 4 years ago
- Deep Learning library for Python. Convnets, recurrent neural networks, and more. Runs on Theano or TensorFlow.☆12Dec 24, 2016Updated 9 years ago
- If it quacks like a tensor...☆60Nov 13, 2024Updated last year
- a pure-python symbolic regression library built on deap☆11Jul 5, 2016Updated 9 years ago
- Adaptive Subgoal Search☆20Apr 3, 2023Updated 2 years ago
- A JAX nn library☆21Sep 9, 2025Updated 6 months ago
- TalkingData AdTracking Fraud Detection Challenge☆10May 8, 2018Updated 7 years ago
- Demo codes in our presentation about MXNet in AWS Seoul Summit 2017☆12Apr 24, 2017Updated 8 years ago
- 1-Click AI Models by DigitalOcean Gradient • AdDeploy popular AI models on DigitalOcean Gradient GPU virtual machines with just a single click and start building anything your business needs.
- 11th Solution of Kaggle TalkingData AdTracking Fraud Detection Challenge☆10May 10, 2018Updated 7 years ago
- Transformer with Mu-Parameterization, implemented in Jax/Flax. Supports FSDP on TPU pods.☆32Jun 5, 2025Updated 9 months ago
- Identifying the language of input text using character-level n-grams, with support for 45 languages☆11Dec 26, 2022Updated 3 years ago
- Einsum-like high-level array sharding API for JAX☆34Jul 16, 2024Updated last year
- ☆18Oct 17, 2023Updated 2 years ago
- Relative Positional Encoding for Transformers with Linear Complexity☆65Apr 1, 2022Updated 3 years ago
- Python 3.6 and TensorFlow implementation of the AReS and MaRS algorithms☆11Jun 23, 2019Updated 6 years ago