Senior Machine Learning Engineer

Overview

Contribute to scalable machine learning platforms and workflows that enable scientists and researchers across Amgen to build, deploy, and manage AI/ML models, primarily in support of drug discovery and development.

What you'll do

• Deliver AI and ML-enabled applications, with deployed models ranging from classical ML models to natural language processing, protein language models, and large language models.
• Contribute to the development and maintenance of ML platform capabilities, including data pipelines and feature engineering workflows, model training, evaluation, and deployment pipelines, experiment tracking and model registry systems, and model performance evaluations and monitoring.
• Partner in implementation of AI and ML Ops best practices, including CI/CD, infra as code, monitoring, traceability and reproducibility.
• Collaborate with cross-functional teams to transition standards and outcomes from experimentation to production-grade enterprise solutions.
• Build and maintain scalable and efficient, productionized MLOps solutions on cloud platforms.
• Develop and deliver training content and knowledge articles to educate resident scientists on model lifecycle management best practices.

What you'll need

• Bachelor's degree with 4 - 8 years of experience in Computer Science, IT or related field.
• Software programming (Python, version control, test driven development).
• Familiarity with cloud technologies (AWS preferred, Databricks).
• Hands-on experience with AI/ML model training and serving.

Nice to have

• Experience with deployment and maintenance of AI/ML models in production.
• Substantial experience with cloud technologies and platforms, such as Databricks, AWS (preferred), or Azure.
• Experience with AI and MLOps frameworks and tools (MLflow, Kubeflow, Weights & Biases, Terraform, etc.).
• Experience with containerization (Docker, Kubernetes).
• Familiarity with distributed data processing (e.g., Spark).
• Familiarity with data science libraries (scikit-learn, TensorFlow, PyTorch, LangChain).

Details

• Location: Hyderabad, India.