← Search

Jessica Grogan

3 accepted papers

2025

Towards Learning High-Precision Least Squares Algorithms with Sequence Models

ICLR 2025poster

This paper investigates whether sequence models can learn to perform numerical algorithms, e.g. gradient descent, on the fundamental problem of least squares. Our goal is to inherit two properties of standard algorithms from numerical analysis: (1) machine precision, i.e. we want to obtain solutions…

2023

Monarch Mixer: A Simple Sub-Quadratic GEMM-Based Architecture

NeurIPS 2023oral

Machine learning models are increasingly being scaled in both sequence length and model dimension to reach longer contexts and better performance. However, existing architectures such as Transformers scale quadratically along both these axes. We ask: are there performant architectures that can scale…

Cited by 59SourcePDFScholar
2022

Monarch: Expressive Structured Matrices for Efficient and Accurate Training

ICML 2022oral

Large neural networks excel in many domains, but they are expensive to train and fine-tune. A popular approach to reduce their compute or memory requirements is to replace dense weight matrices with structured ones (e.g., sparse, low-rank, Fourier transform). These methods have not seen widespread a…