Machine Learning Engineer, Ad Response Prediction

Based in New York, NY (onsite), the Machine Learning Engineer will design, code, and maintain scalable ML pipelines and online serving systems for ad response prediction within Amazon Ads Sponsored Products and Brands, with a salary range of USD 143,700 - 213,800 per year and a minimum of 2 years of professional software development experience.

Responsibilities

• Drive the technical direction of ML 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; you own what you build

Requirements

• 3+ years of non-internship professional software development experience
• 3+ years of full software development life cycle, 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 and AD&D insurance with option for supplemental life plans
• Employee Assistance Program, mental health support, and 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)