Applied Scientist, RL post-training

Amazon • Seattle, WA

5d

About The Position

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 (gender diversity)
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.

Requirements

• 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
• Experience in Postgres, or experience that includes strong analytical skills, attention to detail, and effective communication abilities and experience in Linux/RHEL

Nice To Haves

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

Benefits

• 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
• parental leave

What This Job Offers

• Job Type: Full-time
• Career Level: Mid Level
• Education Level: Ph.D. or professional degree