Jax Data Jobs

1 job found

2mo ago

Machine Learning Engineer

Amazon.com Services LLC Boston, MA

Boston, MA

2mo ago Closed

$144k - $194k/yr Full time Artificial Intelligence Engineer Jax

This position is no longer accepting applications (closed Jun 6, 2026).

Job Description

As an on-site Machine Learning Engineer in Boston, MA, on the Alexa AI team at Amazon, you will contribute to LLM training, production deployment, and optimization, collaborating with Applied Scientists and fellow MLEs to advance generative AI capabilities.

Responsibilities

Collaborate with other engineers to explore design options, prototype new technologies, and assess technical feasibility.
Work closely with Applied Scientists to process data and scale machine learning models.
Operate within an Agile/Scrum environment to deliver high quality software.

Requirements

3+ years of non-internship professional software development experience.
2+ years of non-internship design or architecture experience, including design patterns, reliability, and scalability of new and existing systems.
Experience working with PyTorch or JAX software.
Bachelor's degree or foreign equivalent in Computer Science, Engineering, Mathematics, or a related field.

Technologies

PyTorch
JAX
vLLM
SGLang
TensorRT

Benefits

Health insurance
401(k) matching
Paid time off
Parental leave

About the Team

Central Analytics and Research Science (CARS) is an analytics, software, and science unit within Amazon's Alexa AI organization. Its mission is to deliver scalable, reliable evaluation of state-of-the-art Conversational AI, assessing how customers perceive the assistants they interact with—from metrics to software applications to in-depth analyses—helping assistant developers improve their services.

View full job page →

Get Jax Data Jobs alerts

New Jax Data Jobs, daily.