Applied Scientist, RL post-training, AWS

Amazon Web Services (AWS); Seattle (WA); On-site; USD 167,100 - 226,100; Full time

Benefits offered by this job

Health insurance

401(k) matching

Paid time off

RSUs

Job summary

AWS invites applications for a research-focused role working on Reinforcement Learning post-training of frontier LLMs to advance customer service. Join world-class researchers and academics to push the science powering cloud computing.

You will partner with technology and business teams, access vast data and compute resources, and publish your findings at peer-reviewed venues while advancing state-of-the-art in RL for LLMs and tool use.

Qualifications

• PhD, or Master's degree with 6+ years of applied research experience.
• Experience programming in Java, C++, Python or related language.
• Experience with neural deep learning methods and machine learning.

Skills

Java

C++

Python

Deep learning

Education

PhD, or Master's degree with 6+ years of applied research experience

Tools

MxNet

TensorFlow

Job description

Description

Do you want to work on Reinforcement Learning (RL) post-training of frontier Large Language Models (LLMs) to revolutionize customer service? Come join the world class researchers and academics in the AWS AI endeavor, and develop the science that powers countless new businesses in cloud computing!

Description

Do you want to work on Reinforcement Learning (RL) post-training of frontier Large Language Models (LLMs) to revolutionize customer service? Come join the world class researchers and academics in the AWS AI endeavor, and develop the science that powers countless new businesses in cloud computing! AWS, the world-leading provider of cloud services. Our customers bring problems that will give Applied Scientists like you endless opportunities to see your research have a positive and immediate impact in the world. You will have the opportunity to partner with technology and business teams to solve real-world problems, have access to virtually endless data and computational resources, and to world-class engineers and developers that can help bring your ideas into the world. As part of the team, we expect that you will develop innovative solutions to hard problems, and publish your findings at peer reviewed conferences and journals. The scientific topics you are going to work on include, but are not limited to: LLM post-training to improve capabilities particularly for instruction following, reasoning over long context, and tool use, etc.

About The Team

Why AWS

Amazon Web Services (AWS) is the world’s most comprehensive and broadly adopted cloud platform. We pioneered cloud computing and never stopped innovating — that’s why customers from the most successful startups to Global 500 companies trust our robust suite of products and services to power their businesses.

Inclusive Team Culture

Here at AWS, it’s in our nature to learn and be curious. Our employee-led affinity groups foster a culture of inclusion that empower us to be proud of our differences. Ongoing events and learning experiences, including our Conversations on Race and Ethnicity (CORE) and AmazeCon conferences, inspire us to never stop embracing our uniqueness.

Work/Life Balance

We value work-life harmony. Achieving success at work should never come at the expense of sacrifices at home, which is why we strive for flexibility as part of our working culture. When we feel supported in the workplace and at home, there’s nothing we can’t achieve in the cloud.

Mentorship and Career Growth

We’re continuously raising our performance bar as we strive to become Earth’s Best Employer. That’s why you’ll find endless knowledge-sharing, mentorship and other career-advancing resources here to help you develop into a better-rounded professional.

Diverse Experiences

Amazon values diverse experiences. Even if you do not meet all of the preferred qualifications and skills listed in the job description, we encourage candidates to apply. If your career is just starting, hasn’t followed a traditional path, or includes alternative experiences, don’t let it stop you from applying.

Basic Qualifications

• PhD, or Master's degree and 6+ years of applied research experience
• Experience programming in Java, C++, Python or related language
• Experience with neural deep learning methods and machine learning

Preferred Qualifications

• Experience with popular deep learning frameworks such as MxNet and Tensor Flow.
• First authored publications on the topic of LLM post-training in top-tier AI conferences such as Neurips, ICML, ACL, etc.

Amazon is an equal opportunity employer and does not discriminate on the basis of protected veteran status, disability, or other legally protected status.

Our inclusive culture empowers Amazonians to deliver the best results for our customers. If you have a disability and need a workplace accommodation or adjustment during the application and hiring process, including support for the interview or onboarding process, please visit https://amazon.jobs/content/en/how-we-hire/accommodations for more information. If the country/region you’re applying in isn’t listed, please contact your Recruiting Partner.

The base salary range for this position is listed below. Your Amazon package will include sign-on payments and restricted stock units (RSUs). Final compensation will be determined based on factors including experience, qualifications, and location. Amazon also offers comprehensive benefits including health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance and option for Supplemental life plans, EAP, Mental Health Support, Medical Advice Line, Flexible Spending Accounts, Adoption and Surrogacy Reimbursement coverage), 401(k) matching, paid time off, and parental leave. Learn more about our benefits at https://amazon.jobs/en/benefits.

USA, WA, Seattle - 167,100.00 - 226,100.00 USD annually

Company

- Amazon.com Services LLC

Job ID: A10420019