Note: The job is a remote job and is open to candidates in USA. Optum is a global leader in health care innovation, developing cutting-edge solutions to improve health systems. The Lead AI/ML Software Engineer will own the delivery of machine learning insights, manage scalable systems, and partner with cross-functional teams to implement AI innovations in real-world products.

Responsibilities

• Own the inference and delivery of machine learning insights to multiple customers
• Manage, monitor, and scale machine learning systems, delivering insights to multiple customers
• Partner with cross-functional engineering teams to implement scalable, flexible pipelines that bring AI innovations into real-world products
• Work with large-scale data and distributed computing frameworks to process and analyze high-volume healthcare datasets efficiently
• Drive pipeline efficiency and reliability through profiling, refactoring and advanced engineering
• Develop new data pipelines to improve ML model usage

Skills

• Bachelor of Science in computer science or related quantitative field
• 7+ years of professional software engineering experience, with at least 5+ years building and operating production ML systems
• Expert-level Python experience, including writing production-grade, testable, and maintainable code
• Hands-on experience building and deploying APIs using FastAPI (or similar Python web frameworks) in a production environment
• Solid experience deploying, managing, and scaling services in Kubernetes, including: Containerization (Docker), Helm or equivalent deployment tooling, Resource tuning, autoscaling, and failure recovery
• Proven experience with MLOps practices, including: Model deployment and versioning, Monitoring model performance, drift, and system health
• Experience with AzureML, DataBricks, and Azure Kubernetes
• Experience with distributed computing approaches
• Expertise in ML Ops, for example, model monitoring, drift detection, and event-based model retraining
• Proven solid technical communication skills
• Proven ability and drive to learn independently
• Proven problem-solving and analytical thinking
• Proven ability to explain technical decisions to stakeholders
• All employees working remotely will be required to adhere to UnitedHealth Group's Telecommuter Policy

Benefits

• A comprehensive benefits package
• Incentive and recognition programs
• Equity stock purchase
• 401k contribution (all benefits are subject to eligibility requirements)

Company Overview