Selected Publications

Please checkout my Google Scholar profile for a complete list of my publications.

Nnsight and ndif: Democratizing access to foundation model internals

Jaden Fiotto-Kaufman, Alexander R Loftus, Eric Todd, Jannik Brinkmann, Caden Juang, Koyena Pal, Can Rager, Aaron Mueller, Samuel Marks, Arnab Sen Sharma, Francesca Lucchetti, Michael Ripa, Adam Belfki, Nikhil Prakash, Sumeet Multani, Carla Brodley, Arjun Guha, Jonathan Bell, Byron Wallace, David Bau

ICLR 2025

PDF Project Page

Locating and Editing Factual Associations in Mamba

Arnab Sen Sharma, David Atkinson, David Bau

COLM 2024

PDF Project Page Code

Function Vectors in Large Language Models

Eric Todd, Millicent L Li, Arnab Sen Sharma, Aaron Mueller, Byron C Wallace, David Bau

ICLR 2024

PDF Project Page Code

Linearity of Relation Decoding in Transformer LMs

Evan Hernandez^*, Arnab Sen Sharma^*, Tal Haklay, Kevin Meng, Martin Wattenberg, Jacob Andreas, Yonatan Belinkov, David Bau
^* Equal contribution

ICLR 2024 (Spotlight!)

PDF Project Page Code

Mass-Editing Memory in a Transformer

Kevin Meng, Arnab Sen Sharma, Alex Andonian, Yonatan Belinkov, David Bau

ICLR 2023 (top 25% paper)

PDF Project Page