Loading...
Senior Machine Learning Engineer, GenAIUX
Amazon
H1B ✓On-sitesenior levelPosted April 13, 2026
About the Role
The role involves designing and building end-to-end generative AI systems and integrating them with AWS design platforms. You will lead technical architecture decisions, mentor junior engineers, and optimize AI-driven interfaces for performance and scalability.
Requirements
Candidates must have at least 5 years of professional software development experience, including expertise in machine learning and system architecture. A strong background in leading engineering teams and working with generative or agentic AI technologies is required.
Full Job Description
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 sound judgment, effectively manage stress and work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the Company’s reputation. Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.
Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.
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
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 sound judgment, effectively manage stress and work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the Company’s reputation. Pursuant to the Los Angeles County Fair Chance Ordinance, we will consider for employment qualified applicants with arrest and conviction records.
Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.
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