Hengrui Zhang

Hengrui Zhang 张恒睿

I am a master’s student at the School of Mathematical Sciences, Peking University. I work on optimization and efficient training of large language models, with a particular interest in optimization for machine learning and machine learning for optimization.

I am advised by Prof. Kun Yuan at CMLR, Peking University. Previously, I received my B.S. in Mathematics and Applied Mathematics from Sichuan University in 2026.

Featured project

Selected publications & preprints

2026 Preprint

ReasFlow: Assisting Reasoning-Centric Scientific Discovery in Applied Mathematics via a Knowledge-Based Multi-Agent System

Yutong He*, Daibo Li*, Guohong Li*, Jiahe Geng*, Zhengyang Huang*, Can Ren*, Zekun Zhang, Yifan Liu, Shuchen Zhu, Hengrui Zhang, Boao Kong, Ming Sun, Shu Li, Chenyi Li, Jiang Hu, Kun Yuan, Zaiwen Wen, Pingwen Zhang

A collaborative agent system for mathematical research, from literature synthesis and theoretical reasoning to experiments.

* Equal contribution.

Education & experience

Education

2026 — Present

Peking University

M.Sc. in Big Data
School of Mathematical Sciences

2022 — 2026

Sichuan University

B.S. in Mathematics and Applied Mathematics
School of Mathematics
GPA: 3.90 / 4.0 · Rank: 1 / 40

Research experience

Jul 2024 — Present

AI for Mathematics · BICMR

Mathematical formalization and theorem proving in Lean 4. Outstanding Participant Award and Special Prize.

2025 — Present

Optimization research · Peking University

Working with Prof. Kun Yuan on randomized subspace methods, compositional error propagation, and OptBound.

Feb — May 2025

Optimization-based RLHF

Research with Prof. Ju Ren at Tsinghua University on optimization formulations and convergence in RLHF.

Selected honors

Blog & notes

Real Analysis

Course notes on sets, Lebesgue measure, measurable functions, integration, differentiation, and Lp spaces.

Mathematics · Course notes · In Chinese