Note: The job is a remote job and is open to candidates in USA. Optum AI is UnitedHealth Group’s enterprise AI team focused on developing AI/ML solutions and an enterprise AI/ML development platform for healthcare. The Senior AI/ML Engineer will design and deploy generative AI, NLP, LLM, retrieval-augmented generation, and agentic AI solutions while ensuring production-grade reliability, performance, safety, and governance. The role also involves productionizing AI/ML pipelines and collaborating with product, platform, and data science teams.
Responsibilities
- Design and deploy generative AI and NLP solutions using transformer-based architectures for enterprise use cases such as search, summarization, extraction, conversational AI, and decision support
- Develop and fine-tune large language model (LLM) applications using commercial and open-source models such as OpenAI, Anthropic/Claude, Gemini, LLaMA, or Mistral
- Design and implement retrieval‑augmented generation (RAG) pipelines including document ingestion, chunking strategies, embedding generation, and vector database integration
- Build agentic AI systems including single‑agent and multi‑agent workflows capable of tool use, reasoning, planning, and autonomous task execution
- Implement evaluation and monitoring frameworks for generative AI systems including hallucination detection, bias monitoring, and human‑in‑the‑loop evaluation
- Develop safety and governance controls for AI systems including prompt hardening, policy enforcement, and responsible AI guardrails
- Productionize AI/ML pipelines using CI/CD, testing, monitoring, and observability best practices in cloud‑native environments
- Collaborate with product, platform, and data science teams to translate requirements into scalable AI solutions
Skills
- 5+ years of experience in machine learning, AI engineering, or applied data science including development of production ML systems
- 5+ years of experience developing Python-based ML systems using frameworks such as PyTorch, TensorFlow, or Hugging Face
- 3+ years of experience building Generative AI or LLM applications including prompt engineering, model fine‑tuning, and LLM API integrations
- 2+ years of experience designing retrieval‑augmented generation (RAG) systems and vector search pipelines
- 1+ year of experience building agentic AI systems including tool‑calling, planning frameworks, or multi‑agent workflows
- 1+ year of experience using AI‑assisted development or 'vibe coding' tools such as Codex, Claude Code, Cursor, Windsurf, or similar tools
- Healthcare Experience
- Must be authorized to work in the United States without the need for current or future employer-sponsored visa sponsorship (e.g., H-1B, TN, F-1/OPT, CPT, or other employment-based visa status)
- Experience with LLM orchestration frameworks such as LangChain, LlamaIndex, or Semantic Kernel
- Experience optimizing LLM latency, cost, and performance in production environments
- Experience deploying AI systems on cloud platforms such as AWS, Azure, or GCP
- Experience working in regulated environments with responsible AI governance requirements
Company Overview