← Search

Jinwei Yao

2 accepted papers

2025

DeFT: Decoding with Flash Tree-attention for Efficient Tree-structured LLM Inference

ICLR 2025spotlight

Large language models (LLMs) are increasingly employed for complex tasks that process multiple generation calls in a tree structure with shared prefixes of tokens, including few-shot prompting, multi-step reasoning, speculative decoding, etc. However, existing inference systems for tree-based applic…

2025

ResearchTown: Simulator of Human Research Community

ICML 2025poster

Large Language Models (LLMs) have demonstrated remarkable potential in scientific domains, yet a fundamental question remains unanswered: Can we simulate human research communities with LLMs? Addressing this question can deepen our understanding of the processes behind idea brainstorming and inspire…