Jiarui Wang

I’m a first-year Ph.D. student in Computer Science at the School of Computer Science, Shanghai Jiao Tong University (SJTU), advised by Prof. Zhouhan Lin in Language Understanding and Machine Intelligence Algorithms (LUMIA) Group. I also earned my Bachelor’s degree in Computer Science from SJTU.

My research interests center around machine learning and large language models (LLMs), with recent work centering on:

Parametric Memory — designing and training parametric memory modules that retain domain-specific or pre-training knowledge, without relying on retrieval-augmented generation (RAG)
Memory-augmented LLMs — developing architectures that decouple knowledge and reasoning in LLMs to mitigate catastrophic forgetting and hallucination
Large-scale multi-node vector search — building new multi-machine FAISS-based vector search methods to support scalable memory systems

I’m always open to discussing research, collaboration, and new ideas. Please feel free to reach out!

🔥 News

2025.10 🚀 MLP Memory released — model weights and code are now open-sourced.
2025.09 🚀 Memory Decoder released — model weights and code are now open-sourced.
2025.09 🎉 Memory Decoder accepted to NeurIPS 2025
2024.05 🏆 Our Robocup team SRC won Second Place in RoboCup China Open

Memory Decoder: A Pretrained, Plug-and-Play Memory for Large Language Models
Jiaqi Cao*, Jiarui Wang*, Rubin Wei, Qipeng Guo, Kai Chen, Bowen Zhou, Zhouhan Lin
NeurIPS 2025 Poster
🔗 arXiv · GitHub · HuggingFace · * Equal Contribution
MLP Memory: A Retriever-Pretrained Memory for Large Language Models
Rubin Wei*, Jiaqi Cao*, Jiarui Wang, Jushi Kai, Qipeng Guo, Bowen Zhou, Zhouhan Lin
Preprint (2025)
🔗 arXiv · GitHub · HuggingFace · * Equal Contribution

2025.09–Present, Ph.D. Student, School of Computer Science, Shanghai Jiao Tong University
2021.09–2025.06, B.E. in Computer Science, Shanghai Jiao Tong University

Second Place, RoboCup China Open - Small Size League, 2024
Third Prize, Prototype System Competition, CCF Chinasoft, 2023
Provincial First Prize, Senior Group, National Olympiad in Informatics in Provinces (NOIP), 2018