>_ WeChat
WeChat QR code for adding Max Cao as a friend

CV

Education

Sep 2024 – Present

Shanghai Jiao Tong University · M.E.

  • GPA 3.79 / 4.0
  • Expected graduation: Mar 2027
Sep 2020 – Jun 2024

Shanghai Jiao Tong University · B.E.

  • GPA 3.86 / 4.0
  • Dean’s Scholarship · Rank 1/86 · $7,500

Publications

  1. 2026

    Nanbeige4.2-3B: Unlocking Agentic Capabilities in a Compact Model Tech Report

    Core Contributor

  2. 2026

    Memory Decoder at Scale: A Pretrained, Parametric Long-Term Memory Preprint

    Rubin Wei , Jiaqi Cao , Jiarui Wang , Junming Zhang , Qipeng Guo , Bowen Zhou , Zhouhan Lin

  3. 2026

    MLP Memory: A Retriever-Pretrained Memory for Large Language Models ICLR

    Rubin Wei* , Jiaqi Cao* , Jiarui Wang , Jushi Kai , Qipeng Guo , Bowen Zhou , Zhouhan Lin

  4. 2025

    Memory Decoder: A Pretrained, Plug-and-Play Memory for Large Language Models NeurIPS

    Jiaqi Cao* , Jiarui Wang* , Rubin Wei , Qipeng Guo , Kai Chen , Bowen Zhou , Zhouhan Lin

* Equal contribution · selected works and links on Research.

Experience

Jul 2026 – Present

IQuest Research · Pre-train Intern

Pre-training research for foundation models.

Mar 2026 – Jun 2026

NBG LLM Lab · Pre-train/Post-train Intern

Model arch research and agentic capabilities.

Apr 2025 – Aug 2025

Shanghai AI Lab · Research Intern

Parametric memory and continual learning for LLMs.

Mar 2024 – Aug 2024

Microsoft · Research Intern

AKS diagnostic copilot and RAG retrieval optimization.

Awards & Honors

  • Sep 2023Dean’s Scholarship · Rank 1/86 · highest departmental scholarship · $7,500
  • Apr 2023VEX Robotics World Championship Champion · Team Captain
  • Jan 2021Mathematical Contest in Modeling (MCM) · Meritorious Winner
  • Nov 2018National Olympiad in Informatics in Provinces (NOIP) · First Prize

Skills

  • LanguagesEnglish (TOEFL 109) · French (B2)
  • ProgrammingPython · C/C++
  • ML stackPyTorch · Transformers · Megatron · Slime