Senior Machine Learning Engineer, WebIR - ML Infrastructure

Description

Unlock the future of AI at Amazon AGI (Artificial General Intelligence). At Amazon, we’re at the forefront of transformative AI, shaping the next generation of intelligent technologies. For over 25 years, we’ve developed state-of-the-art AI solutions that transform how businesses serve their customers. Today, as AI stands ready to reshape society, we’re pushing beyond current breakthroughs in generative AI toward the next frontier.

Join our team of scientists, engineers, and experts to help define the future of artificial intelligence. AGI is dedicated to pushing the boundaries of what’s possible, using Amazon’s unparalleled ML infrastructure, computing resources, and commitment to responsible AI principles. We’re looking for the brightest minds from a wide range of backgrounds and experiences to help create transformative AI solutions that will improve lives, solve global challenges, and open up new realms of possibility, from reinventing commerce and accelerating enterprise productivity to advancing universal agents and shaping the future of robotics.

What will you do: you will work with a multidisciplinary team across multiple programs to:

• (i) build and automate training data generation: build a data pipeline for producing high‑quality training data sets for our web information retrieval and ranking models, directly impacting our search quality. Improve data quality by mining hard negatives, incorporating dimensions of quality (relevance, content freshness, page trustworthiness, etc.), and scaling the pipeline to billions of examples. Work closely with scientists to address their specific modeling needs and develop metrics to communicate progress on data quality and scale.
• (ii) accelerate experimental velocity: develop leveraged systems to enable the team to experiment faster. Centralize evaluation workflows and develop tools and systems to streamline production model optimization and deployment.
• (iii) improve system understandability: develop advanced analytics and automate failure‑space analysis processes to help the team debug and understand search quality issues. Partner with the broader AGI analytics team to coordinate metrics for our information retrieval engine with user signals or downstream dependencies to debug across systems as well.
• (iv) push model performance to limits: optimize model inference to maximize hardware utilization, reduce GPU inference latency, and balance trade‑offs with quality for performance.

• 5+ years of non‑internship professional software development experience
• 5+ years of programming with at least one software programming language
• 5+ years of leading design or architecture of new and existing systems (design patterns, reliability and scaling)
• Experience as a mentor, tech lead or leading an engineering team
• 3+ years of experience with technologies such as AWS, Glue, Redshift, Airflow, Spark, Kafka, Kubernetes, Redis

• 5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience
• Bachelor’s degree in computer science or equivalent
• Experience designing or architecting new and existing systems (design patterns, reliability and scaling)
• Experience in machine learning, data mining, information retrieval, statistics or natural language processing
• 2+ years of building large‑scale machine‑learning infrastructure for online recommendation, ads ranking, personalization or search experience
• 2+ years of experience with technologies such as Sage Maker, Triton, PyTorch, ONNX, MLFlow, Flyte

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit (Use the "Apply for this Job" box below). for more information. If the…