Senior AI/ML Engineer

About the role

This is more than a job. It's a chance to shape the future of healthcare through the transformative power of AI.

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

Requirements

• 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)

Qualifications

• 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

Benefits

Learn more about our range of benefits designed to help you live well.

Pay

Pay is based on several factors including but not limited to local labor markets, education, work experience, certifications, etc. In addition to your salary, we offer benefits such as, a comprehensive benefits package, incentive and recognition programs, equity stock purchase and 401k contribution (all benefits are subject to eligibility requirements).

Schedule

All employees working remotely will be required to adhere to UnitedHealth Group's Telecommuter Policy.