Sr AI/ML Engineer

ML Engineer

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 looking for an ML Engineer with solid hands-on experience in GenAI and agentic frameworks, backed by solid foundational ML/NLP knowledge. The role involves building production-grade AI systems across verticals such has Health, Rx, Productization etc.

Primary Responsibilities:

• Build and deploy GenAI and agentic AI solutions in production environments
• Design ML systems involving structured/unstructured data and complex database interactions
• Develop scalable Python-based ML pipelines and APIs
• Work on foundational ML use cases including prediction, optimization, and recommendation systems
• Collaborate with product, engineering, and platform teams to operationalize models
• Take lead in productionizing models including API design and integration, load testing, HPC compute, Latency management
• Comply with the terms and conditions of the employment contract, company policies and procedures, and any and all directives (such as, but not limited to, transfer and/or re-assignment to different work locations, change in teams and/or work shifts, policies in regard to flexibility of work benefits and/or work environment, alternative work arrangements, and other decisions that may arise due to the changing business environment). The Company may adopt, vary or rescind these policies and directives in its absolute discretion and without any limitation (implied or otherwise) on its ability to do so

Required Qualifications:

• Master's or preferably PhD in Computer Science, Artificial Intelligence, Data Science, or a related field
• Solid foundation in machine learning, deep learning, and statistical modeling
• Hands-on experience with NLP, LLMs, transformers, and generative AI techniques
• Experience with building and orchestrating AI agents or multi-agent systems
• Experience working with large-scale datasets and distributed computing (e.g., Spark/PySpark)
• Proficiency in Python and ML frameworks/libraries (e.g., PyTorch, TensorFlow, scikit-learn)
• Understanding of model optimization techniques (e.g., FLOPs reduction, quantization, latency tuning)
• Familiarity with MLOps practices, model deployment, and monitoring in production environments
• Knowledge of APIs, microservices architecture, and system integration
• Solid problem-solving skills and ability to work in a fast-paced, collaborative environment

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.