Lead AI/ML Engineer

Posted Yesterday
Be an Early Applicant
Noida, Gautam Buddha Nagar, Uttar Pradesh, IND
In-Office
Senior level
Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
The Role
Build and deploy full-stack AI/ML solutions across the complete lifecycle, including data preparation, modeling, MLOps, monitoring, and production support. Develop GenAI, LLM, RAG, and agentic features, while creating React-based interfaces with streaming responses and intuitive AI-driven components. Collaborate with engineering, product, design, and data teams to deliver scalable, explainable, secure, and user-friendly AI products.
Summary Generated by Built In
Requisition Number: 2390913
Optum is a global organization that delivers care, aided by technology to help millions of people live healthier lives. The work you do with our team will directly improve health outcomes by connecting people with the care, pharmacy benefits, data and resources they need to feel their best. Here, you will find a culture guided by inclusion, talented peers, comprehensive benefits and career development opportunities. Come make an impact on the communities we serve as you help us advance health optimization on a global scale. Join us to start Caring. Connecting. Growing together.
We are seeking an experienced AI/ML Builder for the AIFWD team, with a solid foundation in Analytics (5+ years), Machine Learning (3+ years), hands-on AI development (2+ years), and front-end engineering, particularly React (2+ years). The ideal candidate is passionate about building scalable, full-stack AI/ML solutions - from model and agent logic through to the user-facing interface - and is committed to ongoing support, optimization, and enhancement of deployed use cases.
This role spans the full lifecycle - from problem discovery to deployment, UI delivery, and operational support - across multiple high-impact AIFWD initiatives.
Primary Responsibilities:
  • AI Builder
    • Design, develop, and deploy AI-powered solutions to solve complex business challenges, with an emphasis on responsible AI use
    • Streamline workflows and automate tasks using enterprise-approved AI tools and ML/LLM frameworks
    • Evaluate emerging AI/ML trends to inform solution design and drive continuous improvement
    • Support the end-to-end lifecycle of AI/ML use cases - data preparation, feature engineering, modeling, deployment, monitoring, MLOps, and post-production enhancements
    • Design and build Gen AI/LLM-powered features such as chat interfaces, copilots, RAG-based search, and agentic workflows
    • Collaborate with cross-functional teams across engineering, product, and data teams to operationalize AI workloads efficiently
  • UI/Frontend
    • Build responsive, production-grade UIs (primarily in React) that expose AI/ML and LLM capabilities to end users
    • Integrate front-end applications with AI/ML backends, APIs, and streaming LLM responses (e.g., token streaming, chat UX, status/progress states)
    • Translate AI/ML outputs (predictions, embeddings, agent actions, RAG citations) into intuitive, usable interface components
    • Collaborate with design/product teams to ensure AI-driven features are explainable, trustworthy, and user-friendly
    • Contribute to component libraries/design systems used across AIFWD AI products
  • Development & Support
    • Build production-ready ML pipelines and services (batch/real-time)
    • Provide L2/L3 support for deployed AI/ML systems, ensuring reliability, model drift management, and performance optimization
    • Conduct root-cause analysis for production incidents involving model performance, data changes, or UI/integration issues
    • Maintain documentation, versioning, and traceability of models, training datasets, and front-end codebases
  • Comply with all applicable Company policies, procedures, and business directives, changes including those relating to work location, team assignments, work schedules, and flexible work arrangements

Required Qualifications:
  • Experience:
    • 5+ years in Data Analytics, statistical modeling, or data-driven problem solving
    • 3+ years in Machine Learning model development (supervised, unsupervised, NLP, forecasting, optimization, etc.)
    • 2+ years in hands-on AI/LLM development (prompt engineering, RAG, agent workflows, embeddings, vector stores, generative models)
    • 2+ years in front-end development, primarily React (component architecture, state management, API integration)
  • Technical Skills:
    • AI/ML & Gen AI
      • Experience with LLM-based development, including instruction tuning, prompt engineering, retrieval-augmented generation (RAG), or evaluation frameworks
      • Practical experience with AI development tools such as Copilot Studio, GitHub Copilot, or M365 Copilot integrations
      • Experience with vector databases/embedding stores (e.g., Pinecone, FAISS, Chroma, Azure AI Search)
      • Solid Python programming experience (NumPy, Pandas, Scikit-Learn, PyTorch/TensorFlow)
      • Knowledge of MLOps tools (Azure ML, AWS Sagemaker, MLflow, Databricks, KubeFlow, Feature Stores)
      • Exposure to agentic AI patterns, automated workflows, and API-based integrations
      • Familiarity with LLM orchestration frameworks (e.g., LangChain, LlamaIndex, Semantic Kernel)
      • Understanding of model evaluation, guardrails, and hallucination mitigation techniques for Gen AI applications
    • Front-End / UI:
      • Solid hands-on experience with React (hooks, component design, state management - Redux/Context/Zustand)
      • Experience building UIs that consume streaming APIs (e.g., SSE/WebSockets for LLM token streaming)
      • Working knowledge of TypeScript, HTML5, CSS3, and modern build tooling (Vite, Webpack)
      • Familiarity with REST/GraphQL API integration and basic UX principles for AI-driven interfaces
      • Exposure to design systems/component libraries (e.g., Material UI, Tailwind, shadcn/ui)
    • Platform & Data:
      • Experience with cloud platforms (Azure preferred) and containerized deployments (Docker/Kubernetes)
      • Familiarity with data engineering fundamentals - ETL/ELT, SQL, data modeling, streaming pipelines
      • Basic understanding of CI/CD pipelines covering both model and front-end deployment

Preferred Qualifications:
  • Experience supporting multiple simultaneous AI/ML use cases, including user-facing AI products
  • Experience as a full-stack AI engineer, bridging ML/LLM backend work with React front-end delivery
  • Understanding of Responsible AI principles and enterprise AI safety guidelines
  • Solid problem-solving orientation and ability to convert ambiguous business problems into well-structured, end-to-end AI solutions (model + UI)
  • Ability to coach teams on using AI tools and contribute to AI adoption across the organization

At UnitedHealth Group, our mission is to help people live healthier lives and make the health system work better for everyone. We believe everyone - of every race, gender, sexuality, age, location and income - deserves the opportunity to live their healthiest life. Today, however, there are still far too many barriers to good health which are disproportionately experienced by people of color, historically marginalized groups and those with lower incomes. We are committed to mitigating our impact on the environment and enabling and delivering equitable care that addresses health disparities and improves health outcomes - an enterprise priority reflected in our mission.

Skills Required

  • 5+ years of experience in data analytics, statistical modeling, or data-driven problem solving
  • 3+ years of experience developing machine learning models
  • 2+ years of hands-on AI or LLM development experience
  • 2+ years of front-end development experience, primarily with React
  • Experience with LLM development, including instruction tuning, prompt engineering, RAG, or evaluation frameworks
  • Experience with Copilot Studio, GitHub Copilot, or Microsoft 365 Copilot integrations
  • Experience with vector databases or embedding stores
  • Solid Python programming experience with NumPy, Pandas, Scikit-Learn, PyTorch, or TensorFlow
  • Knowledge of MLOps tools including Azure ML, AWS SageMaker, MLflow, Databricks, Kubeflow, or feature stores
  • Exposure to agentic AI patterns, automated workflows, and API integrations
  • Familiarity with LLM orchestration frameworks such as LangChain, LlamaIndex, or Semantic Kernel
  • Understanding of model evaluation, guardrails, and hallucination mitigation
  • Hands-on React experience with hooks, component design, and state management
  • Experience building UIs consuming streaming APIs such as SSE or WebSockets
  • Working knowledge of TypeScript, HTML5, CSS3, and modern build tools
  • Familiarity with REST or GraphQL API integration and basic UX principles
  • Exposure to design systems or component libraries
  • Experience with cloud platforms, preferably Azure, and Docker or Kubernetes
  • Familiarity with ETL/ELT, SQL, data modeling, and streaming pipelines
  • Basic understanding of CI/CD pipelines for model and front-end deployment
  • Experience supporting multiple simultaneous AI/ML use cases and user-facing AI products
  • Experience as a full-stack AI engineer bridging ML/LLM backends and React front ends
  • Understanding of Responsible AI principles and enterprise AI safety guidelines
  • Ability to translate ambiguous business problems into end-to-end AI solutions
  • Ability to coach teams on AI tools and contribute to organizational AI adoption

What the Team is Saying

Optum Compensation & Benefits Highlights

  • Leave & Time Off Breadth — Paid time off is often regarded as generous, with meaningful accrual and flexibility to use it. While practices can vary by team, the overall time-off offering stands out as a relative strength.
  • Retirement Support — A 401(k) plan with employer matching is available and generally well-regarded for supporting long-term savings. It remains valuable, though candidates should verify the current match formula.
  • Parental & Family Support — Programs such as parental leave, family medical leave, and adoption assistance are included as notable elements of the total rewards package. These resources can meaningfully support work-life balance when applicable to the role and location.

Optum Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Eden Prairie, MN
160,000 Employees
Year Founded: 2011

What We Do

Optum, part of the UnitedHealth Group family of businesses, is a global organization that delivers care, aided by technology to help millions of people live healthier lives. The work you do with our team will directly improve health outcomes by connecting people with the care, pharmacy benefits, data and resources they need to feel their best. Here, you will find a culture guided by inclusion, talented peers, comprehensive benefits and career development opportunities. Come make an impact on the communities we serve as you help us advance health optimization on a global scale. Join us to start Caring. Connecting. Growing together. At Optum, we support your well-being with an understanding team, extensive benefits and rewarding opportunities. By joining us, you’ll have the resources to drive system transformation while we help you take care of your future. We recognize the power of connection to drive change, improve efficiency and make a difference in health care. Join a team where your skills and ideas can make an impact and where collaboration is key to creating technology that produces healthier outcomes.

Gallery

Gallery
Gallery
Gallery

Optum Offices

Hybrid Workspace

Employees engage in a combination of remote and on-site work.

Optum has three workplace models that balance the needs of the business and the responsibilities of each role. These models, core on‑site (5 days/week), hybrid (4 days/week) and telecommute or fully remote, vary by country, role and location.

Typical time on-site: Not Specified
HQEden Prairie, MN
Metro Manila, Philippines
Cebu, Philippines
Davao, Philippines
Ann Arbor, MI
Atlanta, GA
Baltimore, MD
Bengaluru, India
Chennai, India
Dallas, TX
Detroit, MI
Dublin, Ireland
Hartford, CT
Houston, TX
Hyderabad, India
Jacksonville, FL
Las Vegas, NV
Letterkenny, Ireland
Louisville, KY
Madison, WI
Minneapolis, MN
Nashville, TN
New Delhi, India
Philadelphia, PA
Phoenix, AZ
Pune, India
Raleigh, NC
San Diego, CA
Washington, DC
Learn more

Similar Jobs

Optum Logo Optum

Software Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Noida, Gautam Buddha Nagar, Uttar Pradesh, IND
160000 Employees

Optum Logo Optum

Principal Data Scientist

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Noida, Gautam Buddha Nagar, Uttar Pradesh, IND
160000 Employees

Optum Logo Optum

Software Engineering Lead

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Noida, Gautam Buddha Nagar, Uttar Pradesh, IND
160000 Employees

Optum Logo Optum

Senior Full-stack Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Noida, Gautam Buddha Nagar, Uttar Pradesh, IND
160000 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account