Principal Machine Learning Engineer (Reinforcement Learning)
Role Overview
Role Summary
Principal-level AIML Engineer responsible for designing and deploying advanced AI/ML systems with a strong focus on Reinforcement Learning (RL) and decision intelligence. Will act as a technical leader driving scalable AI solutions from research to production across enterprise platforms.
Key Responsibilities
Design and develop Reinforcement Learning models (RL, RLHF, multi-agent RL) for real-world decision-making problems
Build and deploy scalable ML pipelines and production AI systems using MLOps best practices
Architect end-to-end AI systems integrating RL with GenAI, LLMs, or agent-based frameworks
Lead development of agent-based / multi-agent AI systems for planning, reasoning, and automation
Translate research concepts into production-grade, reliable ML systems
Partner with data scientists, engineers, and product teams to deliver enterprise AI solutions
Evaluate new AI techniques (RLHF, agentic systems, deep RL) and drive adoption
Mentor engineers and provide technical leadership and architectural guidance
Must-Have Skills
Strong expertise in Reinforcement Learning (Deep RL, Policy Optimization, RLHF)
Hands-on experience building production AI/ML systems at scale
Strong programming in Python
Experience with MLOps (MLflow, Kubeflow, SageMaker, etc.)
Knowledge of Distributed Systems & Cloud (AWS/Azure/GCP)
Experience in model deployment, monitoring, and lifecycle management
Strong understanding of ML/DL algorithms and optimization techniques
Good-to-Have
Experience with multi-agent systems / agentic AI frameworks
Exposure to LLMs, RAG, or Generative AI systems
Experience with simulation environments (Gym, RLlib, etc.)
Background in optimization, control systems, or operations research
Full Job Posting
Job Description
Job Description
ABOUT AMGEN
Amgen harnesses the best of biology and technology to fight the world’s toughest diseases, and make people’s lives easier, fuller and longer. We discover, develop, manufacture and deliver innovative medicines to help millions of patients. Amgen helped establish the biotechnology industry more than 40 years ago and remains on the cutting-edge of innovation, using technology and human genetic data to push beyond what’s known today.
ABOUT THE ROLE
Role Summary
Principal-level AIML Engineer responsible for designing and deploying advanced AI/ML systems with a strong focus on Reinforcement Learning (RL) and decision intelligence. Will act as a technical leader driving scalable AI solutions from research to production across enterprise platforms.
Key Responsibilities
Design and develop Reinforcement Learning models (RL, RLHF, multi-agent RL) for real-world decision-making problems
Build and deploy scalable ML pipelines and production AI systems using MLOps best practices
Architect end-to-end AI systems integrating RL with GenAI, LLMs, or agent-based frameworks
Lead development of agent-based / multi-agent AI systems for planning, reasoning, and automation
Translate research concepts into production-grade, reliable ML systems
Partner with data scientists, engineers, and product teams to deliver enterprise AI solutions
Evaluate new AI techniques (RLHF, agentic systems, deep RL) and drive adoption
Mentor engineers and provide technical leadership and architectural guidance
Must-Have Skills
Strong expertise in Reinforcement Learning (Deep RL, Policy Optimization, RLHF)
Hands-on experience building production AI/ML systems at scale
Strong programming in Python
Experience with MLOps (MLflow, Kubeflow, SageMaker, etc.)
Knowledge of Distributed Systems & Cloud (AWS/Azure/GCP)
Experience in model deployment, monitoring, and lifecycle management
Strong understanding of ML/DL algorithms and optimization techniques
Good-to-Have
Experience with multi-agent systems / agentic AI frameworks
Exposure to LLMs, RAG, or Generative AI systems
Experience with simulation environments (Gym, RLlib, etc.)
Background in optimization, control systems, or operations research
Qualifications
- Bachelor’s or Master’s in Computer Science, AI, ML, or related field
- 12 to 17 years experience in ML/AI engineering or related domains
- Behavioral / Leadership Expectations
- Strong technical leadership and mentoring capability
- Ability to translate complex AI concepts into business impact
- Ownership mindset with end-to-end delivery focus
- Collaboration across global teams
- EQUAL OPPORTUNITY STATEMENT
- Amgen is an Equal Opportunity employer and will consider you without regard to your race, color, religion, sex, sexual orientation, gender identity, national origin, protected veteran status, or disability status.
- We will ensure that individuals with disabilities are provided with reasonable accommodation to participate in the job application or interview process, to perform essential job functions, and to receive other benefits and privileges of employment. Please contact us to request an accommodation.
About Amgen
Developing innovative medicines to fight serious diseases.
Visit company websiteJobs and hiring trendsApply for this job in 1 click
Skip the repetitive application forms
Install the Base Career Chrome Extension and autofill job applications across major job boards with your profile.
Trusted by over 500,000 job seekers on Base Career
More from this employer
More jobs at Amgen
Medical Science Liaison - Obesity and Related Conditions - Knoxville,TN
Knoxville, USA
Key Account Manager – Bone Health - Santa Barbara
Santa Barbara, USA
Key Account Manager – Bone Health - Cleveland, OH
Cleveland, USA
Medical Science Liaison - Obesity and Related Conditions - Knoxville,TN
Knoxville, USA
Specialist QA
West Greenwich, USA
Key Account Manager – Bone Health - Santa Barbara
Santa Barbara, USA
Administrative Coordinator
Thousand Oaks, USA
Associate IS Engineer
Hyderabad, IND
Biostatistics Manager - HTA
Uxbridge, GBR
Key Account Manager – Bone Health - Cleveland, OH
Cleveland, USA
Key Account Manager – Bone Health - Memphis, TN
Memphis, USA