Welcome to my personal website! My name is An Xu.

I’m currently an applied scientist at Amazon.com working on shopping LLM. I received my Ph.D degree in Electrical and Computer Engineering from University of Pittsburgh where I worked with Dr. Heng Huang. Prior to that, I received my B.Eng in Electrical Engineering from Tsinghua University.

Feel free to reach out via email (annnxu [at] amazon [dot] com) for collaboration!

We are hiring applied scientist interns on long context LLM.

Research Interests

• large language models: code generation, efficient fine-tuning and inference, sparse MoE LLM
• large-scale machine learning: communication-effcient data parallelism, pipenline parallelism, optimization
• federated learning

Experiences

• Applied Scientist at Amazon.com, Seattle, WA. 2025.1 - now
• Senior Research Scientist at TikTok, Bellevue, WA. 2025.1 - 2025.1
• Research Scientist at TikTok, Bellevue, WA. 2023.6 - 2025.1
• Applied Scientist Intern at Amazon Web Services, Santa Clara, CA. 2022.6 - 2022.8
• Applied Research Intern at NVIDIA Research, remote. 2021.6 - 2021.9

Education

• Ph.D in Electrical and Computer Engineering, University of Pittsburgh, Pittsburgh, PA, U.S., 2023.
• B.Eng in Electrical Engineering (first degree), Tsinghua University, Beijing, China, 2017.
• B.Eng in Business Administration (second degree), Tsinghua University, Beijing, China, 2017.

Academic Services

I have served as a program committee/reviewer for most machine learning conferences.

• ICLR 2022-2025, ICML 2022-2024, NeurIPS 2022-2024
• CVPR 2021-2023, ICCV 2021-2023, ECCV 2020-2024
• AAAI 2022-2025, IJCAI 2023
• KDD 2020, 2023, CIKM 2023

And some journals such as TNNLS and TMI. I was a session chair at ICML 2022.