Jie Zhu

Jie Zhu

Machine Learning Engineer II

@ Amazon

Boston, MA, United Statesjiezwork@gmail.comjiez19@vt.edu
Download CV

About Me

I am a Machine Learning Engineer II at Amazon AGI Customization Service, passionate about advancing AI technologies and building scalable systems. My research focuses on improving the efficiency and effectiveness of large language models, with particular emphasis on training techniques, fine-tuning, distillation, and inference optimization. I have extensive experience in developing enterprise-scale systems, from HCM platforms handling millions of users to voice AI infrastructure powering Alexa experiences.

Research Interests

Artificial Intelligence
Machine Learning
Large Language Models
Inference Deployment
Test-Time Optimization

Work Experience

Machine Learning Engineer II

Amazon AGI Customization Service

Aug 2025 - Present
  • Research cutting-edge LLMs and develop Generative AI customization solutions
  • Nova LLM Training techniques and optimizations
  • Fine-tuning and distillation capabilities
  • Multimodal LLMs leveraging heterogeneous data sources
  • Developing high-efficiency deployment platforms for LLMs

Software Development Engineer II

Amazon People Experience Technology

Dec 2023 - Aug 2025
  • Developed new features for the world's largest HCM system handling million-scale hiring
  • Designed new hiring model for M&A, reducing acquisition process from weeks to hours
  • Optimized hiring document process, reducing paperwork by 80%
  • Improved contingent worker onboarding, reducing start-date delay rate by 20%
  • Developed GenAI solutions: AI self-correction for error resolution, people search tool, and HR chatbot
  • Reduced consultation time from days to hours

Education

M.Eng. in Computer Engineering (Machine Learning)

Virginia Polytechnic Institute and State University

Sep 2019 - Dec 2020

IEEE-Eta Kappa Nu (HKN) Member

Publications

Plan and Budget: Effective and Efficient Test-Time Scaling on Large Language Model Reasoning

J Lin, X Zeng, J Zhu, S Wang, J Shun, J Wu, D Zhou

ICLR 2026Click to view on Google Scholar

Gradient Boosting Decision Tree with LSTM for Investment Prediction

C Yu, F Liu, J Zhu, S Guo, Y Gao, Z Yang, M Liu, Q Xing

Asia-Pacific Conference on Communications Technology and Computer Engineering, 2025Click to view on Google Scholar

Resistive Switching Characteristics of Resistive Random Access Memory Based on a BaxSr1-xTiO3 Thin Film Grown by a Hydrothermal Method

J Zhu, J He, J Lu, C Ma, X Tao, W Liu, Z Feng, L Chang, Y Li, Z Hu

IEEE Electron Device Letters 40 (9), 1411-1414, 2019Click to view on Google Scholar

Negative Differential Resistance and Multilevel Resistive Switching in BaSrTiO3 Films

J He, J Zhu, C Ma, J Lu, Z Hu

Applied Physics Letters 115 (7), 2019Click to view on Google Scholar

Modeling of the Temperature Profiles and Thermoelectric Effects in Phase Change Memory Cells

C Ma, J He, J Lu, J Zhu, Z Hu

Applied Sciences 8 (8), 1238, 2018Click to view on Google Scholar
View All Publications on Google Scholar

Honors & Awards

Student Spotlight

VirginiaTech Northern Virginia Center, Falls Church, VA

IEEE-Eta Kappa Nu (HKN)

Honor Society Member, Beta Lambda Chapter

Visit My Blog

Check out my blog for more insights on AI, machine learning, and software engineering.

Read My Blog