Associate Machine Learning Engineer

Responsibilities

• Collaborate with data scientists to develop, train, and evaluate machine learning models
• Build and maintain MLOps pipelines, including data ingestion, feature engineering, model training, deployment, and monitoring
• Leverage cloud platforms such as AWS and Databricks for ML model development and deployment
• Support the deployment and operation of LLM-based workflows, including prompt configurations and RAG pipelines
• Monitor AI and LLM systems for availability, latency, error rates, and quality degradation
• Develop and execute unit and integration tests to ensure software quality
• Create and maintain documentation for software architecture, design, and operations
• Follow established AI governance, security, and compliance standards

Requirements

• 2 to 6 years of experience in Computer Science, IT, or a related field
• Proficiency in Python or R
• Experience with machine learning libraries such as TensorFlow, PyTorch, or Scikit-learn
• Knowledge of MLOps tools like MLflow, Kubeflow, or Airflow
• Experience with big data technologies such as Spark or Hadoop
• Understanding of the LLM inference lifecycle, including prompt execution and retrieval-augmented generation (RAG)
• Strong critical-thinking and problem-solving skills

Preferred Qualifications

• Experience with containerization technologies like Docker and Kubernetes
• Familiarity with monitoring and logging tools such as Prometheus, Grafana, or Splunk
• Knowledge of GenAI tooling, including vector databases and agent frameworks like LangChain
• Exposure to LLM evaluation techniques regarding grounding, faithfulness, and reliability

About the Company

Amgen harnesses the best of biology and technology to fight the world’s toughest diseases. As a biotechnology pioneer, Amgen discovers, develops, and delivers innovative medicines to help millions of patients worldwide. We leverage advanced human genetics and cutting-edge technology to push the boundaries of what is known in medicine.