Senior Machine Learning Engineer, SageMaker Inference

Are you interested in helping AWS customers by building ML-distributed systems to deploy Generative AI models for inferencing at scale? Sage Maker is a fully managed service to prepare data and build, train, and deploy machine learning (ML) models for any use case with fully managed infrastructure, tools, and workflows. Specifically, the Sage Maker Inference Platform enables customers to deploy ML models at scale to run high throughput, low-latency CPU/GPU/Inferentia workloads and monitor the quality of the models and data in production.

The team is currently focused on solving core infrastructure problems for Generative AI inference serving needs of both internal and external customers. We're moving fast, and this is a great team to join and make a huge impact on AWS especially in the Generative AI space. To learn more about the product, please visit  Key job responsibilities
- Design and implement kernel-level optimizations and inference pipeline improvements that reduce latency and increase throughput for LLM serving workloads.

- Profile, benchmark, and tune model inference performance across hardware and software stacks, identifying bottlenecks and delivering measurable gains.

- Write well-organized, maintainable code and participate actively in code reviews, providing constructive feedback to peers and mentoring newer engineers on the team.

- Collaborate with applied scientists, ML engineers, and infrastructure teams to translate model requirements into production-ready inference solutions.

- Evaluate emerging tools, frameworks, and generative AI techniques for inference optimization, recommending adoption where they improve efficiency or quality.

A day in the life

Here at AWS, we embrace our differences. We are committed to furthering our culture of inclusion. We have ten employee-led affinity groups, reaching 40,000 employees in over 190 chapters globally. We have innovative benefit offerings, and host annual and ongoing learning experiences, including our Conversations on Race and Ethnicity (CORE) and Amaze Con conferences. About the team

SageMaker Inference team is innovating and modernizing the inference experience that enables the customers to customize their inference workloads. This is an opportunity to establish technical vision, shape product strategy, and build within a proven, high-growth service.

Internal job description
* Sage Maker is one of the fastest growing AWS services. You will join the founding team in building an end-to-end ML pipeline platform in Sage Maker. It represents a unique opportunity to contribute in developing a new pillar for Amazon's ML platform.
* Applying ML in real-life is still in its early stage and ML practitioners face similar challenges as software engineering teams faced a decade ago. Our mission is to solve challenges in ML like what Dev Ops solved for software development over the past 10 years. This is a new area in ML with lots of evolvement in concepts, methodology, and solutions.
* We have a deep mix of senior talent presenting itself in Principals and Distinguished Engineers that lead to an extensive level of learning and coaching. This is a great opportunity if you are looking to get into the AI/ML space and want to get hands-on experience building ML platform services at scale.
* Our services are built on native AWS. We embrace modern technologies and server-less/micro-service based architectures, which maximize engineering efficiency and operational excellence.

We own AWS Sage Maker () to empower the world to do more and better Machine Learning. This is recent launch from team made GA in Q2 2026 () We are looking for internal candidates with prior experience building large scale systems, who are excited about getting into the ML domain.

Basic qualifications- 5+ years of non-internship professional software development experience- 5+ years of programming with at least one software programming language experience- 5+ years of leading design or architecture (design patterns, reliability and scaling) of new and existing systems experience
- Experience as a mentor, tech lead or leading an engineering team Preferred qualification - 5+ years of full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations experience

- Experience with CUDA kernels or ML/low-level kernels

- Experience with LLM inference and serving optimization, including kernel tuning (CUDA), latency and throughput tuning,…