Taro Logo

Machine Learning Engineer II, AWS Applied AI Solutions, Visual Reasoning & Agentic AI

Amazon Web Services (AWS) is the world's most comprehensive and broadly adopted cloud platform, pioneering cloud computing and continuous innovation.
$129,300 - $223,600
Machine Learning
Mid-Level Software Engineer
In-Person
5,000+ Employees
3+ years of experience
AI · Enterprise SaaS
This job posting may no longer be active. You may be interested in these related jobs instead:

Description For Machine Learning Engineer II, AWS Applied AI Solutions, Visual Reasoning & Agentic AI

AWS Applied AI Solutions is advancing the frontier of visual reasoning and agentic AI technologies, developing sophisticated AI systems that can understand, interpret, and reason about visual information at human-like levels. As a Machine Learning Engineer II, you'll be at the forefront of developing next-generation visual reasoning systems, working at the intersection of computer vision and large language models.

The role involves tackling fundamental challenges in visual reasoning that combine advanced generative AI and agentic frameworks. You'll be investigating novel approaches to how AI systems understand spatial relationships, reason about object interactions, and maintain contextual awareness across time. Your responsibilities will include implementing agentic AI frameworks, developing token compression techniques, and building efficient inference pipelines.

Working in a collaborative environment, you'll partner with Applied Scientists to advance state-of-the-art while ensuring solutions meet rigorous requirements for accuracy, latency, and cost. The position offers significant opportunities for growth through knowledge-sharing and mentorship, supported by AWS's commitment to continuous innovation and inclusive team culture.

The ideal candidate brings strong software development experience, expertise in machine learning frameworks, and a deep understanding of distributed systems. You'll be part of AWS's mission to pioneer cloud computing and drive innovation, working on cutting-edge AI technologies that power breakthrough applications across multiple industries.

AWS values diverse experiences and provides comprehensive benefits including medical coverage, work-life harmony, and career development opportunities. Join a team that's shaping the future of visual AI technology while enjoying the stability and growth potential of working at the world's leading cloud platform provider.

Last updated 2 months ago

Responsibilities For Machine Learning Engineer II, AWS Applied AI Solutions, Visual Reasoning & Agentic AI

  • Research and implement agentic AI frameworks to solve diverse visual AI challenges
  • Develop novel approaches to token compression and memory management
  • Collaborate with Scientists to enhance multi-modal language models
  • Design and optimize distributed training systems for large-scale visual AI models
  • Build efficient inference pipelines
  • Implement hybrid edge-cloud architectures for deploying visual reasoning systems

Requirements For Machine Learning Engineer II, AWS Applied AI Solutions, Visual Reasoning & Agentic AI

Python
  • 3+ years of non-internship professional software development experience
  • 2+ years of non-internship design or architecture 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
  • Familiar with multi-modal foundation models, pre-training and post-training techniques

Benefits For Machine Learning Engineer II, AWS Applied AI Solutions, Visual Reasoning & Agentic AI

Medical Insurance
Vision Insurance
Dental Insurance
  • Medical benefits
  • Work-life balance
  • Career development and mentorship
  • Employee-led affinity groups
  • Inclusive team culture