Amazon · 17 hours ago
Member of Technical Staff - Reinforcement Learning (Infrastructure), AGI Autonomy
San Francisco, CA
Full-time
Onsite
Mid, Senior Level
$255K/yr - $345K/yr
3+ years expAmazon AGI SF Lab is focused on developing new foundational capabilities for enabling useful AI agents. In this role, you will work closely with research teams to design, build, and maintain systems for training and evaluating state-of-the-art agent models.
Artificial Intelligence (AI)DeliveryE-CommerceFoundational AIRetail
Responsibilities
Develop 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 with programming languages such as Python, Java, C++
Experience with neural deep learning methods and machine learning
Experience with training and deploying machine learning systems to solve large-scale optimizations, or experience troubleshooting and debugging technical systems
Preferred
PhD, or a Master's degree and experience with various machine learning techniques and parameters that affect their performance
Experience with large scale machine learning systems such as profiling and debugging and understanding of system performance and scalability
Experience with distributed system, Megatron, vLLM, Ray, and working with GPUs
Experience with patents or publications at top-tier peer-reviewed conferences or journals
Benefits
Equity
Sign-on payments
Full range of medical, financial, and/or other benefits
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
San Bernardino Sun
2026-02-04
2026-02-04
Company data provided by crunchbase