Machine Learning Engineer, Alexa AI
About the Role
The Alexa AI team is looking for a passionate, talented, and inventive Machine Learning Engineer with a strong machine learning background, to build capabilities such as fine tuning, distillation, and LLM Inference. As a ML engineer with the Alexa AI team, you will be responsible for machine learning platform focus on LLM training, production deployment, and optimizations to advance the state of LLMs. You will collaborate closely with Applied Scientists and other MLEs, leverage Amazon's heterogeneous data sources and large-scale computing resources to accelerate development of Generative Artificial Intelligence solutions.
Key Job Responsibilities
- Will work with other team engineers to investigate design approaches, prototype new technology and evaluate technical feasibility.
- Work closely with Applied scientists to process data, scale machine learning models.
- Will work in an Agile/Scrum environment to deliver high quality software.
About the Team
Central Analytics and Research Science (CARS) is an analytics, software, and science team within Amazon's Alexa AI organization. Our mission is to provide scalable and reliable evaluation of the state-of-the-art Conversational AI on how customers perceive the assistants they interact with – from the metrics themselves to software applications to deep dive on those metrics – allowing assistant developers to improve their services.
Basic Qualifications
- 3+ years of non-internship professional software development experience.
- 2+ years of non-internship design or architecture (design patterns, reliability and scaling) of new and existing systems experience.
- Experience working with PyTorch or JAX software.
- Bachelor's degree or foreign equivalent in Computer Science, Engineering, Mathematics, or a related field.
Preferred Qualifications
- Experience working with PyTorch or JAX software, or experience with vLLM, SGLang, TensorRT or similar platforms in production environments.
- Experience developing large model hosting platforms, establishing frameworks, and scaling and optimizing inference system.
- Experience developing and maintaining MLOps tool in large organizations.
Pay
USA, MA, Boston - 143,700.00 - 194,400.00 USD annually. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location.
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