Machine Learning Engineer, Ad Response Prediction

New York, NY onsite, USD 143,700 to 213,800 per year; Amazon Ads seeks a Machine Learning Engineer to design, implement, and maintain scalable ML pipelines and online serving systems for ad response prediction within the Sponsored Products and Brands organization.

Responsibilities

• Set the technical direction for ML offerings across the Sponsored Products organization.
• Design, implement, debug, and support scalable ML pipelines and online serving systems.
• Collaborate with applied scientists to optimize model performance and the supporting infrastructure.
• Deliver end-to-end solutions with ownership over what you build and operate.

Requirements

• 3+ years of non internship professional software development experience.
• 3+ years of full software development life cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
• 2+ years designing and developing large-scale, multi-tiered, multi-threaded, embedded or distributed software applications, tools, systems, and services using C#, C++, Java, or Perl.
• Knowledge of machine learning model architecture and inference.

Technologies

• C#
• C++
• Java
• Perl
• JAX
• PyTorch
• vLLM
• SGLang
• Dynamo
• TorchXLA
• TensorRT

Benefits

• Health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage)
• 401(k) matching
• Paid time off
• Parental leave
• Sign-on payments
• Restricted stock units (RSUs)

About the Team

The Sponsored Products and Brands team at Amazon Ads is advancing advertising through leading generative AI technologies to help customers discover products and engage with brands across Amazon and beyond.