Senior Machine Learning Engineer, GenAIUX

Help shape new Generative AIUX products for AWS customers

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.

Key job responsibilities
- 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.

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
- Experience with emerging generative and agentic technologies
- Experience in machine learning and data science

Preferred Qualifications

- 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

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Los Angeles County applicants: Job duties for this position include: work safely and cooperatively with other employees, supervisors, and staff; adhere to standards of excellence despite stressful conditions; communicate effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service; and follow all federal, state, and local laws and Company policies. Criminal history may have a direct, adverse, and negative relationship with some of the material job duties of this position. These include the duties and responsibilities listed above, as well as the abilities to adhere to company policies, exercise soun