Code for the paper: Why Transformers Need Adam: A Hessian Perspective
☆63Mar 11, 2025Updated 11 months ago
Alternatives and similar repositories for hessian-spectrum
Users that are interested in hessian-spectrum are comparing it to the libraries listed below
Sorting:
- Code for Adam-mini: Use Fewer Learning Rates To Gain More https://arxiv.org/abs/2406.16793☆453May 13, 2025Updated 9 months ago
- ☆17Oct 30, 2025Updated 4 months ago
- Official implementation for the paper "Controlled Sparsity via Constrained Optimization"☆11Aug 10, 2022Updated 3 years ago
- Official implementation for "Mixture of In-Context Experts Enhance LLMs’ Awareness of Long Contexts" (Accepted by Neurips2024)☆13Jan 7, 2025Updated last year
- [NAACL'25 🏆 SAC Award] Official code for "Advancing MoE Efficiency: A Collaboration-Constrained Routing (C2R) Strategy for Better Expert…☆15Feb 4, 2025Updated last year
- [ICLR 2025] This repository contains the code to reproduce the results from our paper From Sparse Dependence to Sparse Attention: Unveili…☆12Mar 7, 2025Updated last year
- ☆16Apr 26, 2023Updated 2 years ago
- Code for ICLR 2022 Paper (HyperDQN: A Randomized Exploration Method for Deep Reinforcement Learning)☆12Nov 28, 2023Updated 2 years ago
- Inverse Scaling in Test-Time Compute☆25Dec 3, 2025Updated 3 months ago
- Kinetics: Rethinking Test-Time Scaling Laws☆85Jul 11, 2025Updated 7 months ago
- [EMNLP 25] An effective and interpretable weight-editing method for mitigating overly short reasoning in LLMs, and a mechanistic study un…☆17Dec 17, 2025Updated 2 months ago
- [CVPRW 2023] "Many-Task Federated Learning: A New Problem Setting and A Simple Baseline" by Ruisi Cai, Xiaohan Chen, Shiwei Liu, Jayanth …☆13Aug 28, 2023Updated 2 years ago
- ☆18Oct 14, 2024Updated last year
- An extention to the GaLore paper, to perform Natural Gradient Descent in low rank subspace☆18Oct 21, 2024Updated last year
- ☆14May 4, 2024Updated last year
- ☆74Dec 7, 2024Updated last year
- Code for reproducing our paper "Low Rank Adapting Models for Sparse Autoencoder Features"☆17Mar 31, 2025Updated 11 months ago
- Offcial Repo of Paper "Eliminating Position Bias of Language Models: A Mechanistic Approach""☆20Jun 13, 2025Updated 8 months ago
- ☆20Apr 16, 2025Updated 10 months ago
- ☆33Jul 9, 2025Updated 8 months ago
- LISA: Layerwise Importance Sampling for Memory-Efficient Large Language Model Fine-Tuning☆36Apr 4, 2024Updated last year
- ☆15Feb 21, 2024Updated 2 years ago
- Second-Order Fine-Tuning without Pain for LLMs: a Hessian Informed Zeroth-Order Optimizer☆23Feb 11, 2025Updated last year
- Core Utilities for NVIDIA Merlin☆19Jul 26, 2024Updated last year
- ☆20Dec 14, 2024Updated last year
- APOLLO: SGD-like Memory, AdamW-level Performance; MLSys'25 Oustanding Paper Honorable Mention☆271Nov 29, 2025Updated 3 months ago
- Code for the paper "The Journey, Not the Destination: How Data Guides Diffusion Models"☆25Dec 12, 2023Updated 2 years ago
- ☆29Feb 7, 2025Updated last year
- ☆20Nov 4, 2025Updated 4 months ago
- Using PyTorch autograd to compute Hessian of Perplexity for Large Language Models☆27Apr 17, 2025Updated 10 months ago
- [ICML 2025] From Low Rank Gradient Subspace Stabilization to Low-Rank Weights: Observations, Theories and Applications☆52Oct 30, 2025Updated 4 months ago
- ☆21Updated this week
- ☆253Dec 2, 2024Updated last year
- Collection of algorithms for approximating Fisher Information Matrix for Natural Gradient (and second order method in general)☆143May 26, 2019Updated 6 years ago
- Benchmarking Optimizers for LLM Pretraining☆54Dec 30, 2025Updated 2 months ago
- Experimental repository for research implementation of NoLoCo.☆30Jun 15, 2025Updated 8 months ago
- ☆29Nov 9, 2025Updated 4 months ago
- ☆159Updated this week
- DCPO: Dynamic Adaptive Clipping for RL☆47Dec 20, 2025Updated 2 months ago