CV
Education, experience, skills, and awards.
Contact Information
| Name | Zirui Song |
| Professional Title | Machine Learning Engineer |
| zrsong@amazon.com |
Professional Summary
Machine Learning Engineer at AWS OpenSearch, specializing in information retrieval, neural sparse search, and applied deep learning.
Experience
-
Jan 2025 - Present Shanghai, China
Machine Learning Engineer
AWS OpenSearch
- Contributed to Neural Sparse ANN search by integrating the SEISMIC algorithm into OpenSearch, achieving sub-3ms retrieval with 17-35x latency reduction and 18x throughput improvement
- Evaluated and benchmarked leading sparse retrieval and ANN algorithms; identified optimal candidates for production deployment
- Executed memory and performance optimizations (LRU caching, document ID reordering, SIMD), reducing memory footprint by 40%
- Contributed to Neural Sparse Embedding model fine-tuning for improved retrieval quality
-
May 2024 - Aug 2024 Beijing, China
Machine Learning Engineer Intern
Xiaomi AI Lab
- Developed an intermediate fusion multimodal model for predicting mechanical properties of automotive die-cast components
- Built automated data pipeline; deployed SQL database on Kubernetes for cross-team multimodal data access
- Applied semi-supervised learning to leverage unlabeled data, reducing MAE and MAPE by over 50%
-
Sep 2022 - May 2023 Hong Kong
Research Assistant
ROMI Lab, HKPolyU
- Developed a deep neural network (ResNet18 + deconvolution) for organ detection in ultrasound images (80% accuracy)
- Designed robotic arm controller for automated ultrasound scanning with impedance control
-
May 2022 - Aug 2022 Remote
Research Assistant
Sharc Lab, Georgia Tech
- Researched Federated Learning algorithms; optimized multi-device learning based on FedAvg
- Applied low-rank compression to reduce communication costs; proposed new client selection mechanism
Education
-
Aug 2023 - Dec 2024 Ithaca, NY
-
Aug 2019 - May 2023 Hong Kong
The Hong Kong Polytechnic University
Bachelor of Engineering in Electronic and Information Engineering
First-Class Honors
Awards
-
2022 HKSAR Government Scholarship - Reaching Out Award
The HKSAR Government
Awarded for outstanding academic achievement and participation in overseas exchange programs.
-
2022 Hong Kong, China - APEC Scholarship
The HKSAR Government
Awarded for academic excellence in the Asia-Pacific Economic Cooperation region.
Skills
Programming: Python, Java, C++, Rust, SQL
ML & Frameworks: PyTorch, TensorFlow, Keras, Pandas, OpenSearch, Lucene
Infrastructure: Docker, Kubernetes, Linux, Git, AWS
Languages
English : Fluent
Mandarin : Native
Cantonese : Conversational