Member of Technical Staff - Software Engineer - ML Infrastructure, AGI Autonomy

Leading technology company developing AI solutions and cloud services
$120,000 - $350,000
Machine Learning
Staff Software Engineer
In-Person
5,000+ Employees
5+ years of experience
AI · Enterprise SaaS

Description For Member of Technical Staff - Software Engineer - ML Infrastructure, AGI Autonomy

Amazon has launched an innovative research lab in San Francisco focused on developing foundational capabilities for useful AI agents. The lab combines large language models (LLMs) with reinforcement learning (RL) to tackle complex challenges in reasoning, planning, and world modeling across virtual and physical environments. As part of Amazon's broader AGI organization, which recently introduced Amazon Nova, this small but talent-dense team operates with significant autonomy while leveraging Amazon's extensive resources and scale.

The role offers an opportunity to work on cutting-edge AI technology in a fast-paced environment where you'll contribute directly to AI agent development. You'll be developing industry-leading tools for model training, evaluation, and other key research workflows. The position requires strong software development skills and offers exposure to advanced machine learning concepts and GPU infrastructure optimization.

The compensation is highly competitive, ranging from $120,000 to $350,000 based on location and experience, plus additional benefits and potential equity. This is an excellent opportunity for experienced engineers who want to work at the forefront of AI development while having the backing of one of the world's leading tech companies. The role combines the agility of a research lab with the resources and impact potential of Amazon's global scale.

Working in San Francisco, you'll be part of a team that's helping to redefine what's possible with AI, particularly in the realm of practical applications that enhance customer productivity and experience. The position offers significant growth potential and the chance to shape the future of AI agent technology.

Last updated 15 hours ago

Responsibilities For Member of Technical Staff - Software Engineer - ML Infrastructure, AGI Autonomy

  • Contribute directly to AI agent development
  • Develop industry-leading tools for model training
  • Develop evaluation and other key research workflows
  • Work across the software development life cycle
  • Participate in code reviews, build processes, testing, and operations

Requirements For Member of Technical Staff - Software Engineer - ML Infrastructure, AGI Autonomy

Python
Java
  • 5+ years of professional software development experience
  • Proficiency in Python, Java, C++, or related language
  • Bachelor's degree in Computer Science or related field (preferred)
  • Applied knowledge of machine learning theory and practice (preferred)
  • Experience with GPU training infrastructure and optimization (preferred)
  • Experience as a tech lead or engineering mentor (preferred)

Benefits For Member of Technical Staff - Software Engineer - ML Infrastructure, AGI Autonomy

Medical Insurance
Vision Insurance
Dental Insurance
401k
Equity
  • Comprehensive medical, dental, and vision insurance
  • 401(k) plan
  • Equity compensation
  • Total compensation package including base pay and benefits
  • Workplace accommodations available

Interested in this job?

Jobs Related To Amazon Member of Technical Staff - Software Engineer - ML Infrastructure, AGI Autonomy

Language Data Scientist III, AWS Q for Business

Senior data science role at AWS focusing on AI language models, data quality, and enterprise solutions, offering competitive compensation and comprehensive benefits.

Software Development Manager - Compiler, AWS Neuron, Annapurna Labs

Lead compiler engineering team at AWS developing optimization algorithms for machine learning accelerators, managing experienced engineers and driving innovation in ML infrastructure.

Senior Applied Scientist, Featured Merchant Algorithm

Senior Applied Scientist position at Amazon working on Featured Merchant Algorithm to develop and maintain critical machine learning models for e-commerce offer selection.

Member of Technical Staff, PMT (Customer), AGI SF Lab

Technical Product Manager role at Amazon's new AI research lab, focusing on developing and implementing AI agent technologies with competitive compensation and benefits.

Sr. Software Development Manager - ML Compiler, AWS Neuron, Annapurna Labs

Senior Software Development Manager position leading ML compiler development for AWS Neuron team, focusing on machine learning accelerators and compiler optimization.