Machine Learning Engineer II, AWS Applied AI Solutions, Visual Reasoning & Agentic AI
Machine Learning Engineer II, AWS Applied AI Solutions, Visual Reasoning & Agentic AI
Posted on :
26-05-2025
Employer Active
1 Vacancy
Job Alert
You will be updated with latest job alerts via emailValid email field required
Send jobs
Share
Send me jobs like this
Job Alert
You will be updated with latest job alerts via emailValid email field required
Send jobs
Job Description
As part of the AWS Applied AI Solutions organization were advancing the frontier of visual reasoning and agentic AI technologies. Our vision is to develop sophisticated AI systems that can understand interpret and reason about visual information at human-like levels enabling breakthrough applications across multiple industries.
The Team
Our research team is tackling fundamental challenges in visual reasoning that combine advanced generative AI and agentic frameworks. Were investigating novel approaches to how AI systems understand spatial relationships reason about object interactions and maintain contextual awareness across time. Working at the intersection of computer vision and large language models were developing theoretical frameworks and practical techniques that advance the state-of-the-art in visual AI.
Key job responsibilities
- Research and implement agentic AI frameworks to solve diverse visual AI challenges in the physical world
- Develop novel approaches to token compression and memory management for processing continuous visual streams
- Collaborate with Scientists to enhance multi-modal language models for improved visual understanding and reasoning capabilities
- Design and optimize distributed training systems for large-scale visual AI models
- Build efficient inference pipelines that can process multiple visual streams within strict resource constraints
- Implement hybrid edge-cloud architectures for deploying visual reasoning systems at scale
A day in the life
As an MLE on our team youll be at the forefront of developing next-generation visual reasoning systems. Youll tackle challenges like designing efficient architectures for processing visual information implementing sophisticated memory management for long-term reasoning and developing novel approaches to maintain AI capabilities while optimizing for real-world constraints. Youll collaborate closely with Applied Scientists to advance the state-of-the-art while ensuring our solutions meet rigorous requirements for accuracy latency and cost.
About the team
Diverse Experiences
AWS values diverse experiences. Even if you do not meet all of the preferred qualifications and skills listed in the job description we encourage candidates to apply. If your career is just starting hasnt followed a traditional path or includes alternative experiences dont let it stop you from applying.
Why AWS
Amazon Web Services (AWS) is the worlds most comprehensive and broadly adopted cloud platform. We pioneered cloud computing and never stopped innovating thats why customers from the most successful startups to Global 500 companies trust our robust suite of products and services to power their businesses.
Inclusive Team Culture
AWS values curiosity and connection. Our employee-led and company-sponsored affinity groups promote inclusion and empower our people to take pride in what makes us unique. Our inclusion events foster stronger more collaborative teams. Our continual innovation is fueled by the bold ideas fresh perspectives and passionate voices our teams bring to everything we do.
Mentorship & Career Growth
Were continuously raising our performance bar as we strive to become Earths Best Employer. Thats why youll find endless knowledge-sharing mentorship and other career-advancing resources here to help you develop into a better-rounded professional.
Work/Life Balance
We value work-life harmony. Achieving success at work should never come at the expense of sacrifices at home which is why we strive for flexibility as part of our working culture. When we feel supported in the workplace and at home theres nothing we cant achieve.
- 3 years of non-internship professional software development experience including coding standards code reviews source control management build processes testing and operations.
- 2 years of non-internship design or architecture (design patterns reliability and scaling) of new and existing systems experience.
- Background in computer vision multimodal foundation model or related fields.
- Proficiency in Python and distributed systems.
- Experience with deep learning frameworks (PyTorch TensorFlow etc.).
- Experience with real-time inference optimization distributed training systems GPU utilization and memory optimization techniques.
- Familiar with multi-modal foundation models pre-training and post-training techniques.
- Masters or PhD degree in Computer Science Machine Learning or related field.
- Experience designing efficient data preprocessing pipelines building and scaling multi-modal model architectures and conducting robust evaluation at scale.
- Deep understanding of visual reasoning architectures and agentic AI systems.
- Experience with edge computing and model optimization for resource-constrained environments
- Track record of contributions to visual AI or multi-modal learning systems.
- Expertise in large-scale video processing and analysis.
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 Companys 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 for more information. If the country/region youre applying in isnt listed please contact your Recruiting Partner.
Our compensation reflects the cost of labor across several US geographic markets. The base pay for this position ranges from $129300/year in our lowest geographic market up to $223600/year in our highest geographic market. Pay is based on a number of factors including market location and may vary depending on job-related knowledge skills and experience. Amazon is a total compensation company. Dependent on the position offered equity sign-on payments and other forms of compensation may be provided as part of a total compensation package in addition to a full range of medical financial and/or other benefits. For more information please visit This position will remain posted until filled. Applicants should apply via our internal or external career site.
The Team
Our research team is tackling fundamental challenges in visual reasoning that combine advanced generative AI and agentic frameworks. Were investigating novel approaches to how AI systems understand spatial relationships reason about object interactions and maintain contextual awareness across time. Working at the intersection of computer vision and large language models were developing theoretical frameworks and practical techniques that advance the state-of-the-art in visual AI.
Key job responsibilities
- Research and implement agentic AI frameworks to solve diverse visual AI challenges in the physical world
- Develop novel approaches to token compression and memory management for processing continuous visual streams
- Collaborate with Scientists to enhance multi-modal language models for improved visual understanding and reasoning capabilities
- Design and optimize distributed training systems for large-scale visual AI models
- Build efficient inference pipelines that can process multiple visual streams within strict resource constraints
- Implement hybrid edge-cloud architectures for deploying visual reasoning systems at scale
A day in the life
As an MLE on our team youll be at the forefront of developing next-generation visual reasoning systems. Youll tackle challenges like designing efficient architectures for processing visual information implementing sophisticated memory management for long-term reasoning and developing novel approaches to maintain AI capabilities while optimizing for real-world constraints. Youll collaborate closely with Applied Scientists to advance the state-of-the-art while ensuring our solutions meet rigorous requirements for accuracy latency and cost.
About the team
Diverse Experiences
AWS values diverse experiences. Even if you do not meet all of the preferred qualifications and skills listed in the job description we encourage candidates to apply. If your career is just starting hasnt followed a traditional path or includes alternative experiences dont let it stop you from applying.
Why AWS
Amazon Web Services (AWS) is the worlds most comprehensive and broadly adopted cloud platform. We pioneered cloud computing and never stopped innovating thats why customers from the most successful startups to Global 500 companies trust our robust suite of products and services to power their businesses.
Inclusive Team Culture
AWS values curiosity and connection. Our employee-led and company-sponsored affinity groups promote inclusion and empower our people to take pride in what makes us unique. Our inclusion events foster stronger more collaborative teams. Our continual innovation is fueled by the bold ideas fresh perspectives and passionate voices our teams bring to everything we do.
Mentorship & Career Growth
Were continuously raising our performance bar as we strive to become Earths Best Employer. Thats why youll find endless knowledge-sharing mentorship and other career-advancing resources here to help you develop into a better-rounded professional.
Work/Life Balance
We value work-life harmony. Achieving success at work should never come at the expense of sacrifices at home which is why we strive for flexibility as part of our working culture. When we feel supported in the workplace and at home theres nothing we cant achieve.
- 3 years of non-internship professional software development experience including coding standards code reviews source control management build processes testing and operations.
- 2 years of non-internship design or architecture (design patterns reliability and scaling) of new and existing systems experience.
- Background in computer vision multimodal foundation model or related fields.
- Proficiency in Python and distributed systems.
- Experience with deep learning frameworks (PyTorch TensorFlow etc.).
- Experience with real-time inference optimization distributed training systems GPU utilization and memory optimization techniques.
- Familiar with multi-modal foundation models pre-training and post-training techniques.
- Masters or PhD degree in Computer Science Machine Learning or related field.
- Experience designing efficient data preprocessing pipelines building and scaling multi-modal model architectures and conducting robust evaluation at scale.
- Deep understanding of visual reasoning architectures and agentic AI systems.
- Experience with edge computing and model optimization for resource-constrained environments
- Track record of contributions to visual AI or multi-modal learning systems.
- Expertise in large-scale video processing and analysis.
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 Companys 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 for more information. If the country/region youre applying in isnt listed please contact your Recruiting Partner.
Our compensation reflects the cost of labor across several US geographic markets. The base pay for this position ranges from $129300/year in our lowest geographic market up to $223600/year in our highest geographic market. Pay is based on a number of factors including market location and may vary depending on job-related knowledge skills and experience. Amazon is a total compensation company. Dependent on the position offered equity sign-on payments and other forms of compensation may be provided as part of a total compensation package in addition to a full range of medical financial and/or other benefits. For more information please visit This position will remain posted until filled. Applicants should apply via our internal or external career site.
Required Experience:
Unclear Seniority
Employment Type
Full-Time
Department / Functional Area
Key Skills
Report This Job
Disclaimer: Drjobpro.com is only a platform that connects job seekers and employers. Applicants are advised to conduct their own independent research into the credentials of the prospective employer.We always make certain that our clients do not endorse any request for money payments, thus we advise against sharing any personal or bank-related information with any third party. If you suspect fraud or malpractice, please contact us via contact us page.
