← Search

Jared Quincy Davis

5 accepted papers

2024

Are More LLM Calls All You Need? Towards the Scaling Properties of Compound AI Systems

NeurIPS 2024poster

Many recent state-of-the-art results in language tasks were achieved using compound systems that perform multiple Language Model (LM) calls and aggregate their responses. However, there is little understanding of how the number of LM calls -- e.g., when asking the LM to answer each question multiple…

Cited by 13SourcePDFScholar
2022

Decentralized Training of Foundation Models in Heterogeneous Environments

NeurIPS 2022accept

Training foundation models, such as GPT-3 and PaLM, can be extremely expensive, often involving tens of thousands of GPUs running continuously for months. These models are typically trained in specialized clusters featuring fast, homogeneous interconnects and using carefully designed software system…

2021

Rethinking Attention with Performers

ICLR 2021oral

We introduce Performers, Transformer architectures which can estimate regular (softmax) full-rank-attention Transformers with provable accuracy, but using only linear (as opposed to quadratic) space and time complexity, without relying on any priors such as sparsity or low-rankness. To approximate s…

2021

Sub-Linear Memory: How to Make Performers SLiM

NeurIPS 2021poster

Transformer architectures have become very popular yet the original implementation requires $O(L^2)$ in serial time and memory as functions of input length $L$. Recent works proposed various linear self-attention mechanisms, scaling only as $O(L)$ for serial computation. We conduct a thorough compl…

2020

Ode to an ODE

NeurIPS 2020poster

We present a new paradigm for Neural ODE algorithms, called ODEtoODE, where time-dependent parameters of the main flow evolve according to a matrix flow on the orthogonal group O(d). This nested system of two flows, where the parameter-flow is constrained to lie on the compact manifold, provides sta…

Cited by 30SourcePDFScholar