Publications

2026

  1. GATES: Self-Distillation under Privileged Context with Consensus Gating
    Alex Stein, Furong Huang, and Tom Goldstein
    arXiv preprint arXiv:2602.20574, 2026

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
    arXiv preprint arXiv:2402.14020, 2024
  3. A Simple Baseline for Predicting Events with Auto-Regressive Tabular Transformers
    Alex Stein, Samuel Sharpe, Doron Bergman, and 5 more authors
    arXiv preprint arXiv:2410.10648, 2024
  4. Running Huge Context Windows On Tiny GPUs
    Monte Hoover, Ryan Synk, Neel Jain, and 5 more authors
    Preprint, 2024
  5. Algorithm Design for Learned Algorithms
    Avi Schwarzschild, Sean Michael McLeish, Arpit Bansal, and 8 more authors
    Preprint, 2024

2023

  1. Neural Auctions Compromise Bidder Information
    Alex Stein, Avi Schwarzschild, Michael Curry, and 2 more authors
    arXiv preprint arXiv:2303.00116, 2023