Coneheads: Hierarchy Aware Attention. - Albert Tseng; Tao Yu; Toni J. B. Liu; Christopher De Sa (NeurIPS, 2023)

Shadow Cones: Unveiling Partial Orders in Hyperbolic Space. - Tao Yu; Toni J. B. Liu; Albert Tseng; Christopher De Sa (CoRR, 2023)

LLMs learn governing principles of dynamical systems, revealing an in-context neural scaling law. - Toni J. B. Liu; Nicolas Boullé; Raphaël Sarfati; Christopher J. Earls (EMNLP, 2024)

Shadow Cones: A Generalized Framework for Partial Order Embeddings. - Tao Yu; Toni J. B. Liu; Albert Tseng; Christopher De Sa (ICLR, 2024)

Density estimation with LLMs: a geometric investigation of in-context learning trajectories. - Toni J. B. Liu; Nicolas Boullé; Raphaël Sarfati; Christopher J. Earls (ICLR, 2025)

Lines of Thought in Large Language Models. - Raphaël Sarfati; Toni J. B. Liu; Nicolas Boullé; Christopher J. Earls (ICLR, 2025)

What's in a prompt? Language models encode literary style in prompt embeddings. - Raphaël Sarfati; Haley Moller; Toni J. B. Liu; Nicolas Boullé; Christopher J. Earls (CoRR, 2025)