JAX implementation of "Fine-Tuning Language Models with Just Forward Passes"
☆19Jun 10, 2023Updated 3 years ago
Alternatives and similar repositories for mezo-jax
Users that are interested in mezo-jax are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Interpreting neural networks via the STREAK algorithm (streaming weak submodular maximization)☆23Oct 20, 2017Updated 9 years ago
- Go ahead and axolotl questions☆11Feb 6, 2024Updated 2 years ago
- ☆10Jul 6, 2021Updated 5 years ago
- This is the official implementation of the ICML 2023 paper - Can Forward Gradient Match Backpropagation ?☆13May 31, 2023Updated 3 years ago
- Mixture of Kalman filter model for spike sorting (using split & merge EM)☆11May 20, 2014Updated 12 years 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.
- PyRSB: a Python interface to the librsb Sparse Matrix library☆20Mar 18, 2022Updated 4 years ago
- Source code for "Taming GANs with Lookahead–Minmax", ICLR 2021.☆15Mar 28, 2021Updated 5 years ago
- Implementations of the algorithms described in the paper: On the Convergence Theory for Hessian-Free Bilevel Algorithms.☆11Nov 1, 2024Updated last year
- ☆32Aug 28, 2020Updated 6 years ago
- Optimization algorithm which fits a ResNet to CIFAR-10 5x faster than SGD / Adam (with terrible generalization)☆14Oct 20, 2023Updated 3 years ago
- Code for ICLR 2022 Paper, "Controlling Directions Orthogonal to a Classifier"☆35Jun 6, 2023Updated 3 years ago
- Experiments with Super-Universal Newton method.☆13Aug 12, 2022Updated 4 years ago
- [FG 2019 Oral] Attribute-Guided Sketch Generation☆10Jul 25, 2021Updated 5 years ago
- ☆12Apr 26, 2022Updated 4 years ago
- 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.
- Code for LaMPP: Language Models as Probabilistic Priors for Perception and Action☆37Apr 3, 2023Updated 3 years ago
- ☆15May 4, 2024Updated 2 years ago
- Gorilla: An API store for LLMs☆10Jun 28, 2024Updated 2 years ago
- PyTorch implementation of LARS (Layer-wise Adaptive Rate Scaling)☆20May 11, 2019Updated 7 years ago
- ☆22Jan 2, 2026Updated 9 months ago
- A domain-specific probabilistic programming language for modeling and inference with language models☆146Apr 29, 2025Updated last year
- This is the code used for the paper "PMGT-VR: A decentralized proximal-gradient algorithmic framework with variance reduction", prepint.☆15Jul 2, 2022Updated 4 years ago
- An implementation of the penalty-based bilevel gradient descent (PBGD) algorithm and the iterative differentiation (ITD/RHG) methods.☆20Feb 13, 2023Updated 3 years ago
- Data Valuation on In-Context Examples (ACL23)☆24Jan 12, 2025Updated last year
- 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.
- A framework for Bayesian optimization of composite functions.☆15Dec 8, 2022Updated 3 years ago
- ☆18Sep 8, 2026Updated last month
- ☆16Apr 9, 2021Updated 5 years ago
- ☆19Dec 12, 2023Updated 2 years ago
- Reinforcement learning with Equinox☆21Mar 4, 2025Updated last year
- Hessian backpropagation (HBP): PyTorch extension of backpropagation for block-diagonal curvature matrix approximations☆22Mar 25, 2023Updated 3 years ago
- Tools for JAX☆51Oct 23, 2026Updated last week
- Parse CL syntactic constructs in s-expression form (possibly represented as e.g. CSTs) and construct syntax trees☆10Nov 4, 2024Updated last year
- Print an image of a cat to the iTerm2 terminal☆14Feb 7, 2017Updated 9 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.
- The source code of "A Comprehensive Survey and Experimental Study of Subgraph Matching: Trends, Unbiasedness, and Interaction"☆17Sep 6, 2024Updated 2 years ago
- Run DeepSeek V4-Flash (~100B MoE) on a 48GB Apple Silicon Mac via SSD expert-streaming — a custom MLX offload engine. ~4.5-5 tok/s.☆10Jul 4, 2026Updated 3 months ago
- PyTorch tool for training with bigger batch size on the GPU☆11Feb 26, 2021Updated 5 years ago
- A Python implementation of word2vec that allows custom sampling strategies☆10Jan 30, 2014Updated 12 years ago
- Pioupiou 🐤, the nano PPL☆11May 25, 2022Updated 4 years ago
- Fairness-aware, explainable AutoML for quantum + classical ML — 21 models, one Optuna search, SHAP & AI reports. uvx quoptuna☆19Oct 3, 2026Updated 3 weeks ago
- Remove duplicates and near-duplicates from text corpora, no matter the scale.☆13Nov 24, 2023Updated 2 years ago