CV

Education, experience, skills, and awards.

Contact Information

Name Zirui Song
Professional Title Machine Learning Engineer
Email 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

    Cornell University
    Master of Engineering in Electrical and Computer Engineering
    GPA: 3.85/4.0
  • 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