Locke Cai

MTS @ Thinky | prev @ xAI, MIT, Cursor, Together.ai

United States

About

Experience

  • Member of Technical Staff at Thinking Machines Lab
    May 2026 - Present · 3 mos

  • Member of Technical Staff at xAI
    Feb 2026 - May 2026 · 4 mos

  • Research Intern at Together AI
    Jun 2025 - Jan 2026 · 8 mos

    Leading research on fundamental methods to train general reasoning models from unlabeled data. Developed and experimenting novel method grounded in Bayesian Inference and Inverse Reinforcement Learning.

  • Massachusetts Institute of Technology (Cambridge, Massachusetts, United States · On-site)
    • Undergraduate Researcher
      Jun 2024 - Jun 2025 · 1 yr 1 mo

      Led research on Reinforcement Learning (RL) methods to transform Large Language Models (LLMs) into powerful sequential decision makers in complex reasoning tasks. Developed scalable RL algorithms to train specialized LLM agents for end-to-end software development.

    • Undergraduate Researcher
      Oct 2023 - Jun 2024 · 9 mos

      Researched Graph Neural Networks (GNNs) and their capabilities on subgraph-level tasks. Developed novel pre-training and transfer learning frameworks to efficiently adapt pre-trained GNNs to subgraph-level tasks.

  • Research Engineering Intern (Cursor) at Anysphere
    Jan 2024 - Feb 2024 · 2 mos

    Designed and built a new production-ready context retrieval system for LLM prompt generation Significantly enhances LLM’s code-generation capabilities by providing relevant snippets of symbol definitions and references in user’s codebase Designed advanced preloading and asynchronous caching mechanisms to achieve low-latency retrieval