Senior Machine Learning Engineer, GenAIUX

Our team builds Generative AIUX products that transform the way customers design. We are focused on the future of design driven development and democratizing innovation for all builders. We deliver a shared platform across multiple products, and we work on AI-powered tools that use Design Systems to build interfaces, and accelerate design, development, and validation workflows. The two products you will be working on are Venue and project Arena. Venue is an AI powered internal prototyping tool that helps AWS builders create new features using the AWS Cloudscape Design system. Arena is an external version of the tool that helps scale this offering to all of Amazon and to external customers with a focus on 0-1 rapid prototyping.

We are seeking a talented and passionate generative/agentic Senior Machine Learning Engineer to join our team and lead the development of cutting-edge AIUX solutions. AIUX is the intersection of Artificial Intelligence and User Experience. AIUX focuses on making sure the user experience is bar raising for a human and simultaneously works well with agents. This includes designing for trust, transparency (letting users know they are interacting with AI), and providing control (allowing users to edit or undo AI actions). It enhances productivity, automates repetitive tasks, and creates more engaging, intuitive, and efficient user experiences.

The ideal candidate will bring expertise in generative and agentic AI and data analysis to create intelligent and transformative AI applications.

In this role, you will work closely with our product, design, and backend teams to build and enhance our generative AI platform. You will have the opportunity to influence the direction of our product and contribute to a culture of innovation and excellence.

Our team is dedicated to supporting new team members. It has a broad mix of experience levels and Amazon tenures, and we’re building an environment that celebrates knowledge sharing and mentorship. We care about your career growth. We try to assign projects and tasks based on what will help each team member develop into a better-rounded engineer and enable them to take on more complex tasks in the future., * Backend Integration (AI-Focused): Integrate with systems to manage asynchronous data flows from LLMs (Large Language Models) and other AI services.

• Optimize Performance: Ensure maximum speed and responsiveness of AI-driven interfaces, particularly when processing large datasets or real-time streaming output.
• Design and own end-to-end systems, spanning backend services, APIs, data storage, and frontend integration.
• Build and evolve GenAI-powered services that support design generation, validation, and automation workflows.
• Contribute to long-term technical vision and architecture decisions.
• Balance short-term delivery with scalable, maintainable system design.
• Work with internal teams to understand needs and translate them into robust solutions.
• Contribute to hiring, mentor junior engineers, and raise the overall engineering bar.

Do you have experience in Team management?, Do you have a Bachelor’s degree?, * 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
• Experience with emerging generative and agentic technologies
• Experience in machine learning and data science, * Bachelor’s degree in computer science or equivalent
• Experience with full software development life cycle, including coding standards, code reviews, source control management, build processes, testing, and operations
• Experience with agentic orchestration
• Prior experience working with AWS applications, a plus
• Experience with Bedrock, Kiro, or Claude a plus

Pulled from the full job description

• AD&D insurance
• Parental leave
• Health insurance
• 401(k) matching
• Paid time off
• Vision insurance
• Dental insurance, The base salary range for this position is listed below. 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. Amazon also offers comprehensive benefits including 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, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits. USA, CA, East Palo Alto - 193,300.00 - 261,500.00 USD annually USA, CA, Santa Clara - 193,300.00 - 261,500.00 USD annually USA, WA, Seattle - 168,100.00 - 227,400.00 USD annually