Amgen
Hyderabad, Telangana, India Posted 2d ago
Amgen is seeking a Reinforcement Learning Engineer or Machine Learning Engineer with experience in reinforcement learning and DevOps/MLOps to join the AI & Data Science team in Hyderabad. You will work with data scientists and engineers to build and scale production ML systems using feedback and measured outcomes.
Key Responsibilities
Partner with data scientists and software engineers to frame business problems as reinforcement learning.
Implement and evaluate reinforcement learning methods such as contextual bandits or policy learning.
Build and scale Python-based ML systems from prototypes to production using frameworks like PyTorch.
Design evaluation methods and analyze results using appropriate metrics and statistical methods.
Work with human and AI-generated feedback and interaction traces to support model improvement.
Implement DevOps/MLOps best practices to build reliable, reproducible ML workflows.
Requirements
Master’s degree with 5+ years or Bachelor’s degree with 7-9+ years of experience in Computer Science, IT or related field.
Solid foundation in machine learning algorithms and techniques.
Experience in MLOps practices and tools such as MLflow, Kubeflow, or Airflow.
Experience in DevOps tools such as Docker and Kubernetes.
Hands-on experience with RL concepts and Python with libraries like TensorFlow or PyTorch.
Preferred Qualifications
Familiarity with RL environments like Gymnasium, Stable-Baselines3, or Ray RLlib.
Experience with big data technologies like Spark or Hadoop.
Familiarity with large language models and natural language processing.