Sr. Machine Learning Engineer, AGI

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’s Foundational AI (FAI) team is working to push the boundaries of Generative AI using Large Language Models (LLMs). In this onsite role in Sunnyvale, you will help build and evaluate reinforcement learning environments, contribute to frontier model research, and apply Amazon’s heterogeneous data sources and large-scale computing resources to accelerate progress in LLMs.

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

What You’ll Do

• 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

Key Qualifications

• 5+ years of non-internship professional software development experience
• 5+ years of programming with at least one software programming language experience
• 5+ years 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

Compensation

Base salary range: USA, CA, Sunnyvale - USD 193,300 - 261,500 per year.

Benefits

• Sign-on payments
• Restricted stock units (RSUs)
• Health insurance (medical, dental, vision, prescription, Basic Life & AD&D insurance, with option for Supplemental life plans)
• EAP and Mental Health Support
• Medical Advice Line
• Flexible Spending Accounts
• Adoption and Surrogacy Reimbursement coverage
• 401(k) matching
• Paid time off
• Parental leave

About the Team

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

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
• 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
• Those duties include the responsibilities listed above and the ability 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, qualified applicants with arrest and conviction records will be considered

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, visit https://amazon.jobs/content/en/how-we-hire/accommodations.