Reinforcement Learning Engineer
Amgen · Hyderabad
Career Category
EngineeringJob 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 Description:
We are seeking a highly skilled Reinforcement Learning Engineer or Machine Learning Engineer with experience in reinforcement learning (RL) and DevOps/MLOps to join the AI & Data Science team. You will work with data scientists and engineers to build and scale production ML systems using feedback and measured outcomes. You will be expected to own and drive forward scoped projects with technical guidance. The role offers the opportunity to apply reinforcement learning and machine learning methods to practical problems, with appropriate human review and safety controls.
Roles & Responsibilities:
- Partner with data scientists and software engineers to frame business problems as reinforcement learning and identify suitable approaches.
- Implement and evaluate reinforcement learning methods, such as contextual bandits, policy learning, or preference-based approaches, with technical guidance, for applications that learn from feedback.
- Build and scale Python-based ML systems from prototypes to production using frameworks such as PyTorch, scikit-learn, or other relevant machine learning libraries.
- Design evaluation methods and analyze results using appropriate metrics, baselines, statistical methods, and offline or online experiments.
- Work with human and AI-generated feedback, evaluation data, and interaction traces to support model improvement and learning from preferences.
- Implement DevOps/MLOps best practices to build reliable, reproducible ML workflows, including experiment tracking, data and model versioning, testing, and deployment.
- Collaborate with product and platform teams to integrate models into AI applications and monitor their performance after deployment.
- Document technical work, communicate findings, and keep current with relevant developments in reinforcement learning and machine learning.
Must-Have Skills:
- Solid foundation in machine learning algorithms and techniques.
- Experience in MLOps practices and tools (e.g., MLflow, Kubeflow, Airflow); Experience in DevOps tools (e.g., Docker, Kubernetes, CI/CD), including hands-on experience building and deploying ML systems.
- Hands-on experience with RL concepts such as rewards, policies, value functions, explore-exploit tradeoff, and RL with human feedback.
- Proficiency in Python and relevant ML libraries (e.g., TensorFlow, PyTorch, Scikit-learn).
- Ability to design experiments, analyze model behavior, and explain technical results clearly; strong problem-solving and collaboration skills; Ability to learn quickly; Excellent communication and interpersonal skills
Good-to-Have Skills:
- Familiarity with RL environments and libraries such as Gymnasium, Stable-Baselines3, Ray RLlib, TorchRL, or Vowpal Wabbit.
- Experience with big data technologies (e.g., Spark, Hadoop), and performance tuning in query and data processing.
- Experience with data engineering and pipeline development.
- Familiarity with large language models, natural language processing, AI agents, or evaluation of generative AI systems.
Education and Professional Certifications
- Master’s degree with 5+ years of experience in Computer Science, IT or related field
OR
- Bachelor’s degree with 7-9 + years of experience in Computer Science, IT or related field
Soft Skills:
- Excellent verbal and written communication skills
- Ability to work effectively with cross-functional and global teams.
- Ability to own and drive scoped projects with technical guidance and regular feedback.
- Curiosity, initiative, and willingness to learn from feedback.
- Ability to organize work and manage multiple priorities.
- Collaborative approach and focus on shared team goals.
- Attention to detail when handling data, experiments, and technical documentation.
- Ability to present findings to technical and nontechnical audiences.
- Strong analytical, debugging, and problem-solving skills.
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.
Objects in your future are closer than they appear. Join us.
careers.amgen.com
As an organization dedicated to improving the quality of life for people around the world, Amgen fosters an inclusive environment of diverse, ethical, committed and highly accomplished people who respect each other and live the Amgen values to continue advancing science to serve patients. Together, we compete in the fight against serious disease.
Ready to Apply for the Job?
We highly recommend utilizing Workday's robust Career Profile feature to complete the application process. A link to update your profile is available when you click Apply. You can then complete your Workday profile in minutes with the “Upload My Experience” functionality to upload an updated copy of your resume or you can simply edit the individual sections of your Career Profile.
.