LLM Post-training
Fine-tuning and distilling LLMs (SFT, DPO, LoRA) that replaced closed models in Heidi Health's production note generation.
Melbourne, Australia
Hi, I'm
Ph.D. in Electrical Engineering · Staff AI Engineer

Promoted to Staff AI Engineer at Heidi Health.
Guest lecture at UniMelb (ELEN90088) on LLM post-training and agents.Slides
Paper accepted to IEEE Trans. on Information Theory.arXiv
Won 2023 Best EEE Ph.D. Thesis Prize at UniMelb!
Paper accepted to NAACL 2025 (Oral, Top 15%).arXiv
Paper accepted to JMLR (Causal Inference).arXiv
Paper accepted to IEEE Trans. on Information Theory.arXiv
Paper accepted to Artificial Intelligence (AIJ) (Transfer Learning).arXiv
I am a Staff AI Engineer at Heidi Health. I build the fine-tuned large language models (LLMs) and automatic speech recognition (ASR) models behind Heidi's clinical scribe, which replaced closed frontier models in production, and I tech-lead the on-premises deployment of our clinical AI stack for hospitals that cannot use the cloud.
I received my Ph.D. in Electrical Engineering from the University of Melbourne, where I published 10+ first/second-author papers in top venues including IEEE Transactions on Information Theory, JMLR, and ACL. My doctoral research focused on information-theoretic analysis of transfer learning and statistical learning theory.
In industry, I have built production-grade ML systems for healthcare — from LLM-powered clinical decision tools to real-time ASR pipelines and scalable data platforms serving 400+ pharmacies.
I also maintain an active research collaboration with the University of Melbourne on NLP and multi-modal AI, with recent work accepted at ACL 2025 and NAACL 2025.
Fine-tuning and distilling LLMs (SFT, DPO, LoRA) that replaced closed models in Heidi Health's production note generation.
Fine-tuning and distilling speech recognition for medical conversations, from English to Korean, Arabic and Urdu.
Running the full clinical AI stack air-gapped on local NVIDIA GPUs, served with vLLM, SGLang, TensorRT and Triton.
Out-of-distribution detection across text and vision in long dialogues, and LLMs that simulate and assess L2 English learners.
Building end-to-end evaluation frameworks, benchmarks and quality metrics for LLM and ASR systems.
Information-theoretic analysis of transfer and statistical learning, published in IEEE TIT, JMLR and AIJ.
Gao R.*, Wu X.*, Kuribayashi T., Ye M., Qi S., Roever C., Liu Y., Yuan Z., Lau J.
Annual Meeting of the Association for Computational Linguistics (ACL 2025), Main Conference
Wu X., Manton J.H., Aickelin U., Zhu J.
IEEE Transactions on Information Theory
Wu X., Gong M., Manton J.H., Aickelin U., Zhu J.
Journal of Machine Learning Research