Senior ML Engineer: Frontier LLMs & RL Gym Dev (Onsite)

Amazon.com Services LLC; Sunnyvale (CA); On-site; USD 193,000 - 262,000; Full time

Benefits offered by this job

Sign-on payments

Restricted stock units (RSUs)

Health insurance

EAP and Mental Health Support

Medical Advice Line

Flexible Spending Accounts

Adoption and Surrogacy Reimbursement

401(k) matching

Paid time off

Parental leave

Job summary

Amazon.com Services LLC in Sunnyvale, CA, is seeking an on-site role on the Foundational AI (FAI) team to push Generative AI with Large Language Models (LLMs). You will build and evaluate reinforcement learning environments, contribute to frontier model research, and leverage Amazon’s data sources and computing resources to accelerate progress in LLMs.

This position sits within a team focused on improving the customer experience through advanced GenAI capabilities.

Qualifications

• 5+ years of non-internship professional software development experience.
• 5+ years of programming with at least one software programming language experience.
• 5+ years of leading design or architecture (design patterns, reliability, and scaling) of new and existing systems.
• Experience as a mentor, tech lead, or leading an engineering team.
• 5+ years of full software development life cycle experience, including coding standards, code reviews, source control management, build processes, testing, and operations.
• Bachelor’s degree in computer science or equivalent.

Responsibilities

• Support the development of RL Gyms.
• Assess the usefulness of RL Gyms for frontier model advancement.
• Build techniques that advance the state of the art with LLMs.
• Support foundational model development in an applied research role.
• Leverage Amazon’s heterogeneous data sources and large-scale computing resources to accelerate advances in LLMs.

Skills

Software development

Programming

Technical leadership

Mentorship

SDLC

Education

Bachelor's degree in computer science or equivalent

Job description

Amazon.com Services LLC in Sunnyvale, CA, is seeking an on-site role on the Foundational AI (FAI) team to push Generative AI with Large Language Models (LLMs). You will build and evaluate reinforcement learning environments, contribute to frontier model research, and leverage Amazon’s data sources and computing resources to accelerate progress in LLMs.

This position sits within a team focused on improving the customer experience through advanced GenAI capabilities.