~/about
Hi, I’m Allan.
I’m a Research Scientist at ByteDance Seed in Singapore, working on LLM agents and reinforcement learning. I’m a core contributor to Seed-Prover, a family of Lean 4 theorem provers trained with large-scale agentic RL — Seed-Prover reached an IMO-certified, silver-medal-level score at IMO 2025, and Seed-Prover 1.5 proves 88% of PutnamBench.
I’m especially interested in agentic RL: long-horizon credit assignment, reward design, and learning from experience.
Before Seed, I was a Research Scientist at Salesforce AI Research, working on efficient long-context inference (ThinK, an ICLR 2025 Spotlight), and an NLP Scientist at ByteDance AI Lab, working on mathematical reasoning with language models — including ReFT, which brings reinforcement learning to chain-of-thought fine-tuning.
I received my Ph.D. from the Singapore University of Technology and Design, advised by Prof. Wei Lu in the StatNLP group, where I built structured models for named entity recognition and semantic parsing; the thesis received SUTD’s Best Thesis Award. Along the way I interned at the Allen Institute for AI and at Alibaba.
01/experience
Experience & education
-
6b801fe (HEAD → main) 2025 — now
Research Scientist @ ByteDance Seed Singapore
LLM agents and reinforcement learning. Core contributor to Seed-Prover — Lean 4 theorem-proving agents trained with large-scale agentic RL (IMO 2025 silver-level score; 88% of PutnamBench).
-
15cf2b7 2024 — 2025
Research Scientist @ Salesforce AI Research Singapore
Efficient long-context inference for LLMs — ThinK, query-driven KV-cache pruning (ICLR 2025 Spotlight).
-
63ee5f5 2020 — 2024
NLP Scientist @ ByteDance AI Lab / ByteDance Research Singapore
Reasoning with language models: ReFT (reinforced fine-tuning, ACL 2024), chain-of-thought design, few-shot prompting for numerical reasoning, and visual document understanding.
-
1f9fee5 2019 — 2020
Research Intern @ Alibaba Singapore
Knowledge-graph-to-text generation (ENT-DESC, EMNLP 2020) and named entity recognition.
-
022f388 2019
Research Intern @ Allen Institute for AI (AI2) Seattle
Hosted by Pradeep Dasigi and Ana Marasović.
-
1e3af2c (tag: phd) 2016 — 2020
Ph.D., Computer Science @ Singapore University of Technology and Design (SUTD)
StatNLP group, advised by Prof. Wei Lu. Thesis: “Leveraging Dependency Trees for Structured Prediction”. ★ Best Thesis Award
02/service
Service
- Senior Program Committee
- AAAI 2023
- Program Committee / Reviewer
- ACL 2020–2022 · EMNLP 2018–2022 · NAACL 2019 · AACL 2022 · AAAI 2019 · NLPCC 2017, 2020 · PACLIC 2018 · IJCNLP 2017
- Journal Reviewer
- IEEE Transactions on Emerging Topics in Computing · Natural Language Engineering · ACM TALLIP
04/open source
Open source
github.com/allanj →allanj/ pytorch_neural_crf
LSTM / BERT-CRF for named entity recognition in PyTorch.
allanj/ repo-level-codegen-papers
A curated list of repository-level code generation papers.
allanj/ ner_incomplete_annotation
NER with incomplete annotations (NAACL 2019).
allanj/ ner_with_dependency
Dependency-guided LSTM-CRF (EMNLP 2019).
allanj/ Deductive-MWP
Deductive reasoner for math word problems (ACL 2022).
allanj/ LayoutLMv3-DocVQA
Fine-tuning LayoutLMv3 on DocVQA.
sutd-statnlp/ statnlp-neural
Hypergraph-based structured prediction framework from StatNLP — I was an active maintainer.
tools/ jsonv
A tiny in-browser JSON visualizer with collapsible trees — handy for inspecting model outputs.