About this role
Amazon Ads seeks a Machine Learning Engineer to own end-to-end ML pipelines for ad response prediction, collaborating with scientists and engineers to optimize models and scale inference for a large product catalog.
Key Responsibilities
- Own technical direction of ML offerings
- Build and optimize ML pipelines for ad prediction
- Collaborate with scientists, designers, and engineers
- Ensure scalable online serving with low latency
- Drive experimentation and model improvements
Technical Overview
Production-grade ML stack requiring ML frameworks (PyTorch, JAX, TensorRT) and experience with transformer architectures, large-scale inference, and AI-assisted development.
Ideal Candidate
The ideal candidate is a hands-on ML engineer with 3+ years of software development experience and 2+ years building large-scale ML systems for ads or recommendations, proficient with transformer models and leading ML frameworks.
Must-Have Skills
3+ years of non-internship professional software development experience3+ years of full software development life cycleincluding coding standardscode reviewssource control managementbuild processestestingand operations experience2+ years designing and developing large-scalemulti-tieredmulti-threadedembedded or distributed software applicationstoolssystemsand services using: C#C++Javaor Perl experienceKnowledge of machine learning model architecture and inference
Nice-to-Have Skills
Knowledge of Machine Learning and LLM fundamentalsincluding transformer architecturetraining/inference lifecyclesand optimization techniquesKnowledge of ML frameworks including JAXPyTorchvLLMSGLangDynamoTorchXLAand TensorRT1+ years building large-scale machine-learning infrastructure for online recommendationads rankingpersonalization or search experienceDemonstrated track record of innovative AI solution development
Tools & Platforms
PyTorchJAXvLLMSGLangDynamoTorchXLATensorRTTransformer architecturesML infrastructure
Required Skills
C#C++JavaPerlPythonPyTorchJAXvLLMSGLangDynamoTorchXLATensorRTTransformer architectureML model architectureInference
Hard Skills
C#C++JavaPerlPythonPyTorchJAXvLLMSGLangDynamoTorchXLATensorRTTransformer architectureML model architectureInference
Soft Skills
analyticalcommunicationteamworkcollaborationownershipbias toward shipping
Keywords for Your Resume
machine learning engineerad response predictionamazon adsai-assisted developmenttransformer architecturetraining/inference lifecyclespytorchjaxvllmsglangdynamotorchxlatensorrtc++javaperlc#pythonlarge-scale ml infrastructureonline recommendationads rankinglatency-sensitiveml frameworksml engineertransformerpyTorchvLLMTorchXLATensorRTPythonC++
Get matched to jobs like this
Luna finds roles that fit your skills and career goals — no endless scrolling required.
Create a Free Profile