Amazon

Member of Technical Staff - Reinforcement Learning (Infrastructure), AGI Autonomy

Amazon$255K — $345K *
Information Technology
Less than 5 years of experience
Job Overview by Ladders

Qualifications

  • PhD or Master's degree with 3+ years of applied research experience
  • Proficient in programming languages like Python, Java, C++
  • Experience with neural deep learning methods and machine learning
  • Skilled in training and deploying machine learning systems for large-scale optimizations
  • Ability to troubleshoot and debug complex technical systems

Responsibilities

  • Develop training infrastructure for large-scale reinforcement learning on LLMs
  • Work across the entire technology stack from ML systems to data management
  • Analyze and troubleshoot complex ML systems, identifying performance bottlenecks
  • Collaborate with researchers on MLSys research to innovate techniques and tooling

Benefits

  • Comprehensive employee benefits package including health and wellness programs
  • Opportunities for professional development and growth
  • Access to cutting-edge technology and resources
  • Inclusive workplace culture that values diversity and innovation
Full Job Description
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.

Our team works inside the Amazon AGI SF Lab, an environment designed to empower AI researchers and engineers to work with speed and focus. Our philosophy combines the agility of a startup with the resources of Amazon.

Key job 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.

BASIC QUALIFICATIONS

- 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 QUALIFICATIONS

- 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.

The base pay for this position ranges from $255,000/year in our lowest geographic market up to $345,000/year in our highest geographic market. Pay is based on a number of factors including market location and may vary depending on job-related knowledge, skills, and experience. Amazon is a total compensation company. Dependent on the position offered, equity, sign-on payments, and other forms of compensation may be provided as part of a total compensation package, in addition to a full range of medical, financial, and/or other benefits. For more information, please visit https://www.aboutamazon.com/workplace/employee-benefits.

About Amazon

Audible is a provider of spoken audio information and entertainment , on the Internet. They provide premium spoken audio content, such as audio versions of books and newspapers and radio programs, that is delivered over the Internet and played back on personal computers and hand-held electronic devices. The Audible service allows consumers to purchase and download their content from their Website, store it in digital files and play it back on personal computers and electronic devices. More than 15,000 hours of audio content are available on their Web site, including audio versions of books, periodicals and radio programs. Several manufacturers have agreed to support and promote the playback of their content on their hand-held audio-enabled electronic devices.

Amazon Careers

Joining Amazon presents an unparalleled opportunity to become part of a vibrant team pushing the boundaries of innovation and growth in the global marketplace. As a leader in e-commerce, technology, and logistics, Amazon offers a variety of job opportunities that cater to a range of skills and professional interests. Work You’ll Do At Amazon, every day is an opportunity to collaborate with the brightest minds in technology and business to redefine what’s possible. Whether you’re interested in software development, marketing, human resources, or customer service, Amazon has a position waiting for you. Transform the way the world shops and innovates with our diverse and inclusive team. Amazon is not just a company; it’s a community where you can drive real change and contribute to projects impacting millions globally. Lead with Innovation and Leadership Amazon is the perfect place to enhance your leadership and innovation skills. Our culture encourages pushing the envelope and imagining the unimaginable. Here, you will lead projects that challenge the status quo and define new industry standards. Work with a team that values diversity and is committed to creating an inclusive environment. Our leadership is focused on harnessing the collective power of unique perspectives to foster growth and innovation. Explore Amazon’s Employment Benefits Amazon’s commitment to its employees extends beyond just career growth. We offer competitive benefits, including health care, parental leave, and diversity training, ensuring that our team not only excels professionally but also enjoys well-being and security. Internship and Networking Opportunities Start your career with an Amazon internship and gain hands-on experience that matters. Our internships provide a gateway to full-time employment and an opportunity to network with professionals across various sectors of the company. Future-Proof Your Career With Amazon, your career path is filled with numerous opportunities for advancement. Our learning and development programs are designed to nurture your professional growth and keep you at the forefront of industry trends. Stay Connected Join Our Team Discover the job opportunities at Amazon that match your skills and interests. We are constantly on the lookout for passionate, curious, and innovative team players ready to make a difference. Keep Up to Date Stay ahead with career tips, insider perspectives, and industry-leading insights you can put to use today—all from the people who work here. Job Alert Emails Customize your subscription to receive job alerts, the latest news, and insider tips tailored to your preferences. Explore the exciting and rewarding career opportunities that await at Amazon. Amazon is more than just a company—it’s a platform for building a promising future. Whether you’re starting or looking to advance your career, Amazon offers the resources, support, and network you need to succeed. Join us, and be a part of our continuing mission to be Earth's most customer-centric company.
Learn more about Amazon
Size
1,608 employees
Market Cap
$832.6 billion
Industry
Net Income
$21.3 billion
Founded
1994
5 Year Trend
+28.1%
Revenue
$386 billion
NASDAQ

Similar Jobs

More Jobs at Amazon

More Information Technology Jobs

Find similar Member of Technical Staff - Reinforcement Learning (Infrastructure), AGI Autonomy jobs: