Machine Learning Engineer II, AWS Applied AI Solutions, Visual Reasoning & Agentic AI
Amazon
DESCRIPTION
As part of the AWS Applied AI Solutions organization, we're 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. We're 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, we're 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, you'll be at the forefront of developing next-generation visual reasoning systems. You'll 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. You'll 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, hasn’t followed a traditional path, or includes alternative experiences, don’t let it stop you from applying.
Why AWS?
Amazon Web Services (AWS) is the world’s most comprehensive and broadly adopted cloud platform. We pioneered cloud computing and never stopped innovating — that’s 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
We’re continuously raising our performance bar as we strive to become Earth’s Best Employer. That’s why you’ll 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, there’s nothing we can’t achieve.