Publications

2024

  1. Transformers Can Do Arithmetic with the Right Embeddings
    Sean McLeish, Arpit Bansal, Alex Stein, and 8 more authors
    In The Thirty-eighth Annual Conference on Neural Information Processing Systems, 2024
  2. Coercing LLMs to do and reveal (almost) anything
    Jonas Geiping, Alex Stein, Manli Shu, and 3 more authors
    2024
    Submitted to IEEE Secure and Trustworthy Machine Learning (SaTML) (Under Review)
  3. A Simple Baseline for Predicting Events with Auto-Regressive Tabular Transformers
    Alex Stein, Samuel Sharpe, Doron Bergman, and 5 more authors
    2024
    Submitted to The Thirteenth International Conference on Learning Representations (ICLR) (Under Review)
  4. Running Huge Context Windows On Tiny GPUs
    Monte Hoover, Ryan Synk, Neel Jain, and 5 more authors
    2024
    Submitted to The Thirteenth International Conference on Learning Representations (ICLR) (Under Review)
  5. Algorithm Design for Learned Algorithms
    Avi Schwarzschild, Sean Michael McLeish, Arpit Bansal, and 8 more authors
    2024

2023

  1. Neural Auctions Compromise Bidder Information
    Alex Stein, Avi Schwarzschild, Michael Curry, and 2 more authors
    2023
  2. What Algorithms do Machines Learn to Solve Mazes?
    Avi Schwarzschild, Arpit Bansal, Sean McLeish, and 2 more authors
    2023
    arXiv pending