Amazon · 22 hours ago
Member of Technical Staff - Reinforcement Learning (Infrastructure), AGI Autonomy
San Francisco, CA
Full-time
Onsite
Mid, Senior Level
3+ years expAmazon is focused on developing new foundational capabilities for enabling useful AI agents through its AGI SF Lab. In this role, you will design, build, and maintain systems for training and evaluating state-of-the-art agent models, working closely with research teams to enhance AI capabilities.
Artificial Intelligence (AI)DeliveryE-CommerceFoundational AIRetail
Responsibilities
Develop cutting-edge training infrastructure to ensure large-scale reinforcement learning on LLMs runs highly efficient and robust
Work across the entire technology stack, including low level ML system, job orchestration and data management
Analyze, troubleshoot and profiling complex ML systems, identify and address performance bottlenecks
Work closely with researchers, conduct MLSys research to create new techniques, infrastructure, and tooling around emerging research capabilities
Qualification
Required
PhD, or Master's degree and 3+ years of applied research experience
Experience programming in Java, C++, Python or related language
Experience with neural deep learning methods and machine learning
Experience debugging ML systems
Preferred
PhD in Computer Science, Machine Learning, or a related field, with a focus on ML System
Demonstrated experience in developing, implementing and debugging large scale ML systems
Experience with distributed system, Megatron, vLLM, Ray, and working with GPUs
Experience with patents or publications at top-tier peer-reviewed conferences or journals
Company
Amazon
Amazon is a tech firm with a focus on e-commerce, cloud computing, digital streaming, and artificial intelligence.
10001+ employees
H1B Sponsorship
Amazon has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (22803)
2024 (21175)
2023 (19057)
2022 (24088)
2021 (12233)
2020 (14881)
Funding
Current Stage
Public CompanyTotal Funding
$8.11BKey Investors
AmazonKleiner Perkins
2023-01-03Post Ipo Debt· $8B
2001-07-24Post Ipo Equity· $100M
1997-05-15IPO
Leadership Team
Douglas J. Herrington
CEO, Worldwide Amazon Stores
Werner Vogels
VP & CTO
Recent News
2026-01-13
2026-01-13
Business Standard India
2026-01-13
Company data provided by crunchbase