Sr. Associate - Machine Learning Engineer

Amgen SA; Hyderabad; On-site; INR 2,600,000 - 4,200,000; Full time

Job summary

Amgen is seeking a hands-on Machine Learning Engineer (Level 4) in Hyderabad to design, build and productionise generative AI systems with safety and auditability. You will own LLM and agent capabilities, evaluation frameworks and guardrails, aligning with data science, product, platform and compliance teams.

You will implement MLOps practices, deploy scalable services in cloud environments with Kubernetes, and partner with security and regulatory stakeholders to ensure enterprise readiness.

Qualifications

• 5–9 years of applied ML or software engineering experience with 2+ years in production ML.
• B.Tech, M.Tech or MS with CS specialization.
• Hands-on experience with large language models: fine-tuning, instruction tuning, retrieval augmentation and embeddings.
• Experience with agent frameworks and multi-step agent behaviours.
• Strong Python and DL framework skills (PyTorch, JAX, TensorFlow).
• Experience building evaluation suites, human-in-the-loop workflows and adversarial testing.
• Guardrails and safety mechanisms for NLP/LLM systems.
• Cloud deployment experience with Kubernetes and model serving.
• Clear communicator capable of cross-functional engagement.

Responsibilities

• Design, implement and deploy LLM solutions and agent frameworks for production workflows.
• Build and run evaluation pipelines: metrics, tests, adversarial tests, and human-in-the-loop eval.
• Create and operationalise guardrails: prompts, policy enforcement, input sanitisation, output validation.
• Prototype and productionise agentic behaviours: RAGs, multi-step planners, tool-use, state management.
• Implement MLOps: versioning, CI/CD for models, scalable serving, observability and cost controls.
• Partner with security, privacy, regulatory and ethics teams for compliance and auditability.

Skills

Applied ML

Production ML

Software engineering

LLMs

Fine-tuning

Instruction tuning

Retrieval augmentation

Embeddings

Agent frameworks

Multi-step agents

Python

PyTorch

JAX

TensorFlow

Kubernetes

Model serving

Communication

Cross-functional collaboration

Education

B.Tech, M.Tech or MS in CS

Tools

PyTorch

TensorFlow

JAX

Kubernetes

Python

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

Amgen is hiring a hands‑on Machine Learning Engineer (Level 4) to design, build and operationalise generative and agentic AI systems that are safe, auditable and production ready. You will own LLM and agent capabilities, evaluation frameworks and guardrails, and work with data science, product, platform and compliance partners to move models from prototype into trusted enterprise services.

ROLE DESCRIPTION:

KEY RESPONSIBILITIES:

• Design, implement and deploy LLM solutions and agent frameworks for production workflows, focusing on reliability, interpretability and safety.
• Build and run evaluation pipelines: automated metrics, unit and integration test suites, adversarial/red‑team tests, and human‑in‑the‑loop evaluation.
• Create and operationalise guardrails: prompt patterns, policy enforcement, input sanitisation, output validation, data filtering and explainability tools.
• Prototype and productionise agentic behaviours: RAGs, multi‑step planners, tool‑use interfaces, state management and safe action execution.
• Implement MLOps best practices: model versioning, CI/CD for models, scalable serving, observability and cost controls.
• Partner with security, privacy, regulatory and ethics teams to embed compliance and auditability into deployments.
• Stay updated with the latest trends and advancements

MUST-HAVE SKILLS:

• 5 to 9 years of applied machine learning or software engineering experience, including at least 2 years on production ML systems.
• B.Tech, M.Tech or MS with specialisation in Computer Science.
• Strong hands‑on experience with large language models: fine‑tuning, instruction tuning, retrieval augmentation and embeddings.
• Practical experience with agent frameworks and multi‑step agent behaviours.
• Solid software engineering skills in Python and deep learning frameworks such as PyTorch, JAX or TensorFlow.
• Experience building evaluation suites, human‑in‑the‑loop workflows and adversarial testing.
• Demonstrated implementation of guardrails and safety mechanisms for NLP/LLM systems.
• Experience deploying services in cloud environments and with container orchestration (Kubernetes) and model serving technologies.
• Clear communicator able to translate prototypes into production solutions and to engage cross‑functional stakeholders.

GOOD-TO-HAVE SKILLS:

• Prior work on Agentic AI, multi‑agent systems, planners or applied agent safety research.
• Experience with benchmarking and continuous evaluation tooling.
• Background in regulated industries such as pharma or healthcare.
• Familiarity with Traditional ML, data governance and model risk management.
• Advanced technical credentialing or proven contributions to open‑source or academic work in LLMs or agent systems.

BEHAVIORS AND COMPETENCIES:

• Pragmatic execution: delivers practical, well‑engineered solutions that work in production.
• Ownership: drives projects from design through sustained operations.
• Security and compliance mindset: designs for auditability, traceability and risk reduction.
• Collaborative: aligns product, platform, governance and data science stakeholders.
• Continuous learner: keeps pace with LLM and agent research and applies best practices.