shikaiqiu / compute-better-spentView external linksLinks
☆63Oct 3, 2024Updated last year
Alternatives and similar repositories for compute-better-spent
Users that are interested in compute-better-spent are comparing it to the libraries listed below
Sorting:
- ☆33Oct 4, 2024Updated last year
- Code for the paper "Function-Space Learning Rates"☆25Jun 3, 2025Updated 8 months ago
- [Oral; Neurips OPT2024 ] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers☆14Mar 18, 2025Updated 10 months ago
- This repository contains the code for the paper "TaylorShift: Shifting the Complexity of Self-Attention from Squared to Linear (and Back)…☆13Mar 17, 2025Updated 11 months ago
- Code for NeurIPS 2024 Spotlight: "Scaling Laws and Compute-Optimal Training Beyond Fixed Training Durations"☆89Oct 30, 2024Updated last year
- train with kittens!☆63Oct 25, 2024Updated last year
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated last year
- GoldFinch and other hybrid transformer components☆45Jul 20, 2024Updated last year
- ☆35Apr 12, 2024Updated last year
- ☆53May 20, 2024Updated last year
- A benchmark dataset and simple code examples for measuring the perception and reasoning of multi-sensor Vision Language models.☆19Dec 27, 2024Updated last year
- ☆19Dec 4, 2025Updated 2 months ago
- ☆40Jan 5, 2024Updated 2 years ago
- ☆24Sep 25, 2024Updated last year
- ☆23Oct 15, 2024Updated last year
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- ☆11May 12, 2023Updated 2 years ago
- ☆11Oct 11, 2023Updated 2 years ago
- ☆47Jan 18, 2024Updated 2 years ago
- Experiments on the impact of depth in transformers and SSMs.☆40Oct 23, 2025Updated 3 months ago
- [ICML 2025] From Low Rank Gradient Subspace Stabilization to Low-Rank Weights: Observations, Theories and Applications☆52Oct 30, 2025Updated 3 months ago
- Repo for "Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture"☆562Dec 28, 2024Updated last year
- Code for "Optimizing DDPM Sampling with Shortcut Fine-Tuning" (https://arxiv.org/abs/2301.13362), ICML 2023☆30Oct 6, 2023Updated 2 years ago
- Accelerated First Order Parallel Associative Scan☆196Jan 7, 2026Updated last month
- Official implementation of "Hydra: Bidirectional State Space Models Through Generalized Matrix Mixers"☆169Jan 30, 2025Updated last year
- GoldFinch and other hybrid transformer components☆12Dec 9, 2025Updated 2 months ago
- recipe for training fully-featured self supervised image jepa models☆12Jun 4, 2025Updated 8 months ago
- ☆14Mar 2, 2025Updated 11 months ago
- ☆12Sep 16, 2024Updated last year
- Combining SOAP and MUON☆19Feb 11, 2025Updated last year
- An annotated implementation of the Hyena Hierarchy paper☆34May 28, 2023Updated 2 years ago
- ☆252Dec 2, 2024Updated last year
- FlexAttention w/ FlashAttention3 Support☆27Oct 5, 2024Updated last year
- 🧱 Modula software package☆321Aug 18, 2025Updated 5 months ago
- Supporting code for the blog post on modular manifolds.☆115Sep 26, 2025Updated 4 months ago
- ☆13Jan 15, 2025Updated last year
- Find context neurons in Pythia models.☆13Jun 13, 2023Updated 2 years ago
- A library for unit scaling in PyTorch☆133Jul 11, 2025Updated 7 months ago
- Code for lin-RFM used for sparse recovery tasks☆15Mar 13, 2025Updated 11 months ago