Jiakang Huang

I am Jiakang (Daniel) Huang, a final-year B.Sc. student in Computer Science at the University of British Columbia. I am currently a visiting researcher in the BigML Lab at UCLA, advised by Prof. Baharan Mirzasoleiman, and previously worked as a research assistant at the University of Toronto.

I believe that as models grow, what matters shifts from how much we spend to where we spend it. Every training run has a fixed budget of data, compute, and parameters, and we mostly spread it evenly, but the return is uneven. I am interested in finding where that budget matters with a measurable signal, drawing on data selection, model merging, optimization, and ML compilers.

At UCLA I work on teacher selection for LLM post-training: using cheap, student-aware signals to pick the right teacher model for reasoning SFT before generating data with every teacher, and extending this kind of data curation to agentic trajectories. I am applying to PhD programs for Fall 2027.

Beyond research, I co-founded iMark, an AI bookmark assistant built on retrieval-augmented generation. Outside of work, I enjoy basketball and CS:GO.

Data Selection LLM Post-Training Model Merging ML Compilers Efficient ML

News

Education

University of British Columbia

Bachelor of Science in Computer Science

Sep. 2022 - May 2027

Peking University

Exchange Student, Computer Engineering

May 2024 - Sep. 2024

Chengdu Foreign Languages School

High School

Sep. 2019 - Jun. 2022

Experience

University of California, Los Angeles

Visiting Researcher @BigML Lab

Jun. 2026 - Current

University of Toronto

Research Assistant

Dec. 2025 - Jun. 2026

University of British Columbia

Undergraduate Teaching Assistant @CPSC 213

Sep. 2024 – Dec. 2024

Publications

NeurIPS 2026 Workshop 2026

HR-Merge: Model Merging with Headroom-Weighted Residual Refinement

Jiakang Huang, Xueyan Zhang, Renyi Cai, Jingcheng Niu, Jiaru Li, Yining Wang, Josh Zhang, Jiahe Jiang, Jinman Zhao, Gerald Penn, Baharan Mirzasoleiman

HR-Merge: Model Merging with Headroom-Weighted Residual Refinement

NeurIPS 2026 NeuralArtifacts Workshop (Poster). A training-free, merge-operator-agnostic wrapper that provably never underperforms the base merge on validation. It gives the best result in every CLIP ViT and LoRA setting tested, reaching 94.29 on the LLaMA-3 8B six-NLI benchmark.

EMNLP 2026 Main 2026

Syntactic Prediction through Reinforcement Learning

Jinman Zhao, Yining Wang, Jiahe Liu, Xueyan Zhang, Linbo Cao, Jiakang Huang, Yiren Zhao, Renyi Cai, YiTian Ding, Gerald Penn

Syntactic Prediction through Reinforcement Learning

Accepted to EMNLP 2026 (Main Conference). Syntax-R1 is a two-stage post-training framework that uses reinforcement learning to improve hierarchical syntactic prediction in large language models, achieving state-of-the-art results on constituency parsing and CCG supertagging.

Under Review 2026

GraphFuse-R1: A Compiler-Side Bound and a Learned Graph-Level Policy for Operator Fusion

Xueyan Zhang, Jiakang Huang, YiTian Ding, Shuhao Guan, Yining Wang, Yuchen Li, Jinman Zhao, Gerald Penn

GraphFuse-R1: A Compiler-Side Bound and a Learned Graph-Level Policy for Operator Fusion

Under review. GraphFuse-R1 trains an 8B LLM with SFT and GRPO to propose whole-graph operator-fusion groups inside PyTorch's TorchInductor, compiling 28.5% fewer kernels and fusing 39% more nodes than the hand-written heuristic, and derives a compiler-side bound on how much fusion headroom remains.

Under Review 2026

TARE: Timescale-Aware Row-wise Energy Redistribution for Muon Optimizers

Jinman Zhao, Xueyan Zhang, Jiakang Huang, Xutao Mao, Jingcheng Niu, Xi Chen, Yining Wang, Changyi Yang, YiTian Ding, Jiahe Liu, Gerald Penn, Jiaru Li

TARE: Timescale-Aware Row-wise Energy Redistribution for Muon Optimizers

Under review. A Muon-family optimizer that shifts update energy away from weight rows whose recent gradients conflict with the longer-term trend, reaching NorMuon's loss with 1.07–1.10x higher token efficiency from 130M to 1.1B parameters at under 5% extra wall-clock time.

Projects

Open Source StarWebsite 2026

OpenTikZ

OpenTikZ demo

An open-source TikZ figure library and AI agent skill for academic papers. Describe a figure, such as a system diagram, neural-network architecture, or pipeline, and get back paper-ready, editable TikZ.

Chrome Extension Website 2025

iMark

iMark demo

Co-founder & AI Technical Lead. An agentic bookmark assistant with semantic search and web interactions, shipped as a Chrome extension with 100+ users. Built on retrieval-augmented generation (Pinecone + Gemini), with Stripe and Supabase.

Web App 2024

UBC Campus Explorer

UBC Campus Explorer demo

A full-stack UBC campus navigation app: a custom query engine over UBC building and room data, with Google Maps walking routes between selected buildings (100+ weekly visits). Built with TypeScript, Express/Node.js, the Google Maps API, and Docker on AWS EC2.

Blog

October 14, 2026

Jiakang Huang

In the AI Era, Iteration Speed Is the Edge

In the AI era, how fast you iterate and how many projects you can move forward have become the skills that matter most. Three habits that speed me up: work like a manager, stay ...

In the AI Era, Iteration Speed Is the Edge