Code for the paper "Function-Space Learning Rates"
☆25Jun 3, 2025Updated 9 months ago
Alternatives and similar repositories for function-space-learning-rates-paper
Users that are interested in function-space-learning-rates-paper are comparing it to the libraries listed below. We may earn a commission when you buy through links labeled 'Ad' on this page.
Sorting:
- Combining SOAP and MUON☆19Feb 11, 2025Updated last year
- Schedule free optimiser implemented in JAX using Optimistix☆15May 29, 2024Updated last year
- ☆68Mar 21, 2025Updated last year
- Minimal (truly) muP implementation, consistent with TP4 and TP5 papers notation☆14Jan 2, 2026Updated 2 months ago
- RWKV-7 mini☆12Mar 29, 2025Updated 11 months ago
- Code for implementing central flows☆44Sep 5, 2025Updated 6 months ago
- Supporting code for the blog post on modular manifolds.☆120Sep 26, 2025Updated 5 months ago
- ☆24Dec 11, 2024Updated last year
- ☆24Jun 4, 2024Updated last year
- Github Repository for the HOI4 ULTRA Project.☆11Updated this week
- [Oral; Neurips OPT2024 ] μLO: Compute-Efficient Meta-Generalization of Learned Optimizers☆16Feb 12, 2026Updated last month
- ☆40Jan 5, 2024Updated 2 years ago
- JAX Scalify: end-to-end scaled arithmetics☆18Oct 30, 2024Updated last year
- ☆56Feb 24, 2026Updated last month
- ☆33Oct 4, 2024Updated last year
- Code for "What really matters in matrix-whitening optimizers?"☆23Oct 31, 2025Updated 4 months ago
- [ICLR 2025] SDTT: a simple and effective distillation method for discrete diffusion models☆47Feb 26, 2026Updated 3 weeks ago
- ☆27May 3, 2024Updated last year
- Code for paper Almost-Orthogonal Layers for Efficient General-Purpose Lipschitz Networks☆12Aug 9, 2022Updated 3 years ago
- Probabilistic Finite Volume Method based on Affine Gaussian Process inference☆11Jun 10, 2024Updated last year
- ☆15Mar 2, 2025Updated last year
- some common Huggingface transformers in maximal update parametrization (µP)☆87Mar 14, 2022Updated 4 years ago
- WIP☆95Aug 13, 2024Updated last year
- [NeurIPS 2025] Official code for "Tropical Attention: Neural Algorithmic Reasoning for Combinatorial Algorithms"☆27Oct 23, 2025Updated 5 months ago
- H-Net Dynamic Hierarchical Architecture☆81Sep 11, 2025Updated 6 months ago
- ☆63Oct 3, 2024Updated last year
- nanoGPT using Equinox☆15Mar 3, 2023Updated 3 years ago
- Twisted client library for AMQP (tested against RabbitMQ). This is a mirror and fork of the launchpad project: https://launchpad.net/txam…☆18May 23, 2012Updated 13 years ago
- An unofficial implementation of "Mixture-of-Depths: Dynamically allocating compute in transformer-based language models"☆36Jun 7, 2024Updated last year
- A single-line modification to any (dualizer-based) optimizer that allows the optimizer to adapt to the scale of the gradients as they cha…☆19Jan 11, 2025Updated last year
- ☆19Dec 4, 2025Updated 3 months ago
- ☆13Jun 3, 2024Updated last year
- ☆25Mar 14, 2026Updated last week
- train with kittens!☆64Oct 25, 2024Updated last year
- 🧱 Modula software package☆324Aug 18, 2025Updated 7 months ago
- A library for unit scaling in PyTorch☆133Jul 11, 2025Updated 8 months ago
- An efficient implementation of the NSA (Native Sparse Attention) kernel☆131Jun 24, 2025Updated 9 months ago
- Simple implementation of muP, based on Spectral Condition for Feature Learning. The implementation is SGD only, dont use it for Adam☆86Jul 28, 2024Updated last year
- Simple Scalable Discrete Diffusion for text in PyTorch☆37Sep 27, 2024Updated last year