Sr. Machine Learning Engineer, AGI

Amazon’s Foundational AI (FAI) team is building Generative AI technology powered by Large Language Models (LLMs), with a focus on pushing frontier capabilities forward. In this onsite role in Sunnyvale, CA, you will contribute to RL Gym development, support frontier model research, and use Amazon’s heterogeneous data sources alongside large-scale computing to accelerate LLM progress.

What you’ll do

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

Basic qualifications

• 5+ years of non-internship professional software development experience.
• 5+ years of programming with at least one software programming language.
• 5+ years leading design or architecture (including design patterns, reliability, and scaling) for 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.

Team mission

The FAI team’s mission is to push the envelope in GenAI with LLMs in order to provide the best-possible experience for customers.

Compensation and location

• Location: Sunnyvale, CA (onsite)
• Base salary range: USD 193,300 to 261,500 per year

Benefits

• Sign-on payments
• Restricted stock units (RSUs)
• 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

Los Angeles County applicants

• Work safely and cooperatively with other employees, supervisors, and staff.
• Adhere to standards of excellence despite stressful conditions.
• Communicate effectively and respectfully with employees, supervisors, and staff to ensure exceptional customer service and follow all federal, state, and local laws and Company policies.
• Criminal history may have a direct, adverse, and negative relationship with some of the material job duties of this position, including the duties and responsibilities listed above, as well as the abilities to adhere to company policies, exercise sound judgment, effectively manage stress and work safely and respectfully with others, exhibit trustworthiness and professionalism, and safeguard business operations and the Company’s reputation.
• Pursuant to the Los Angeles County Fair Chance Ordinance, Amazon will consider for employment qualified applicants with arrest and conviction records.

Inclusive culture

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.