Machine Learning Engineer, Ad Response Prediction

Machine Learning Engineer, Ad Response Prediction at Amazon.com Services LLC, onsite in New York, NY, with a salary range of USD 143,700 - 213,800 per year, responsible for designing, coding, and maintaining scalable ML pipelines and online serving systems for ad response prediction within the Sponsored Products and Brands team.

Responsibilities

• Set the technical direction for machine learning offerings across the Sponsored Products organization
• Design, code, troubleshoot, and support scalable ML pipelines and online serving systems
• Collaborate with applied scientists to optimize ML model performance and infrastructure
• Deliver end-to-end solutions and own what you create

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 experience
• 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 re-imagining the advertising landscape through industry leading generative AI technologies, revolutionizing how millions of customers discover products and engage with brands across Amazon.com and beyond. We are at the forefront of re-inventing advertising experiences, bridging human creativity with artificial intelligence to transform every aspect of the advertising lifecycle from ad creation and optimization to performance analysis and cu