Lead AI/ML Engineer

UnitedHealth Group

2–5 yrs Noida Full Time Hybrid (office + remote)
UnitedHealth Group logo
Posted : today
Actively hiring

Job description

Join Optum, a global leader in health optimization, as an experienced AI/ML Builder. This role is instrumental in developing and deploying scalable, full-stack AI/ML solutions that directly enhance health outcomes. You will be part of a collaborative team focused on innovation, leveraging cutting-edge AI technologies to connect people with the care they need. Our inclusive culture fosters continuous learning and offers ample career development opportunities as we work together to advance global health.

This position offers the chance to work across the entire AI/ML lifecycle, from initial concept and problem discovery through to deployment, user interface delivery, and ongoing operational support. You'll contribute to high-impact initiatives within the AIFWD team, driving advancements in health optimization on a global scale.

Responsibilities

As an AI Builder, you will design, develop, and deploy AI-powered solutions to address complex business challenges, always prioritizing responsible AI principles. Streamline workflows and automate tasks using approved enterprise AI tools and ML/LLM frameworks. Evaluate emerging AI/ML trends to inform solution design and promote continuous improvement.

You will support the complete lifecycle of AI/ML use cases, including data preparation, feature engineering, modeling, deployment, monitoring, MLOps, and post-production enhancements. A key part of this role involves building Gen AI/LLM-powered features like chat interfaces, copilots, and RAG-based search. You will also collaborate with cross-functional engineering, product, and data teams to efficiently operationalize AI workloads.

For UI/Frontend development, you will build responsive, production-grade user interfaces, primarily in React, to present AI/ML and LLM capabilities to end-users. Integrate front-end applications with AI/ML backends, APIs, and handle streaming LLM responses, ensuring intuitive user experiences. Translate AI/ML outputs into usable interface components and work with design/product teams to ensure AI features are explainable and user-friendly.

Development & Support responsibilities include building production-ready ML pipelines and services. You will provide L2/L3 support for deployed AI/ML systems, ensuring reliability and performance optimization. Conduct root-cause analysis for production incidents and maintain comprehensive documentation for models, datasets, and codebases, adhering to all company policies.

Qualifications

We are seeking candidates with a strong foundation in Data Analytics (5+ years), Machine Learning model development (3+ years), hands-on AI/LLM development (2+ years), and front-end engineering, specifically React (2+ years). A passion for building end-to-end AI/ML solutions, from model logic to user interface, is essential.

Required technical skills encompass extensive experience with LLM-based development, including prompt engineering, RAG, and evaluation frameworks. Proficiency in Python (NumPy, Pandas, Scikit-Learn, PyTorch/TensorFlow) is crucial. Experience with MLOps tools and containerized deployments (Docker/Kubernetes) is expected. Familiarity with agentic AI patterns, orchestration frameworks, and model evaluation techniques is also required.

For front-end development, solid hands-on experience with React (hooks, state management) is necessary, along with experience building UIs that consume streaming APIs. Working knowledge of TypeScript, HTML5, CSS3, and modern build tooling is expected. Familiarity with REST/GraphQL API integration and basic UX principles for AI interfaces is important.

Experience with cloud platforms, particularly Azure, and data engineering fundamentals like SQL and data modeling are beneficial. Preferred qualifications include experience supporting multiple AI/ML use cases, acting as a full-stack AI engineer, and a solid understanding of Responsible AI principles. The ability to convert ambiguous business problems into structured, end-to-end AI solutions is highly valued.

Essential Skills

Data AnalyticsMachine LearningAI DevelopmentFront-end EngineeringReactPythonLLMRAGPrompt EngineeringVector DatabasesMLOpsTypeScriptHTML5CSS3REST APIsGraphQLDockerKubernetesSQLCI/CD

Good to Have

AzureAWS SagemakerMLflowDatabricksKubeFlowFeature StoresLangChainLlamaIndexSemantic KernelMaterial UITailwindshadcn/uiETL/ELTStreaming Pipelines

Highlights

  • Actively hiring

More Details

RoleLead AI/ML Engineer
IndustryHealthcare
DepartmentData Science, Product Management
Employment TypeFull Time, Hybrid (office + remote)

About the Company

UnitedHealth logo

UnitedHealth

Healthcare

Lead AI/ML Engineer at UnitedHealth Group | SkillMX | SkillMX