AI Or ML Engineer

UnitedHealth Group; Dadri; On-site; INR 3,500,000 - 7,000,000; Full time

Job summary

UnitedHealth Group is seeking an AI Engineering Lead to architect and deliver enterprise-grade AI solutions across healthcare data ecosystems. You will mentor teams, drive innovation in Generative AI, and ensure accuracy, scalability, and robust design across Azure/GCP environments, with collaboration across product, data science, and engineering.

The role emphasizes responsible AI, production-grade pipelines, and cross-functional leadership to deliver customer-centric solutions for improved

Qualifications

• Bachelor's or Master's in CS, Data Science, Statistics, or related field.
• 5+ years in Data Science, AI/ML, or similar role, with 2+ years in leadership.
• Hands‑on experience delivering production‑grade AI/ML projects.
• Experience with the full lifecycle of AI/ML projects, including EDA, model development, tuning, and drift analysis.
• Experience with Predictive Maintenance, IoT, Sensors, Smart Manufacturing, and Digital Twins.
• Experience with Time series models, anomaly detection, and RUL of equipment.
• Solid programming skills in Python with pandas, numpy, and SQL.
• Proficiency in frameworks such as Scikit Learn, TensorFlow, and PyTorch.

Responsibilities

• Lead design, development, and implementation of scalable AI/ML solutions.
• Oversee predictive maintenance for MRI and CT scanners.
• Manage end-to-end ML workflows from data prep to deployment and monitoring.
• Guide model selection, optimization, and balancing performance and explainability.
• Build production-grade ML pipelines with MLflow, feature stores, and ONNX.
• Ensure ML systems meet scalability, security, reliability, and observability standards.
• Collaborate with data engineering to integrate real-time and batch pipelines.
• Mentor data scientists and ML engineers in best practices and cloud deployment.
• Communicate complex concepts to technical and non-technical stakeholders.

Skills

Python

pandas

numpy

SQL

TensorFlow

PyTorch

Scikit Learn

Leadership

Data Science

ML Deployment

Education

Bachelor's or Master's in CS/DS/Statistics

Tools

MLflow

Feature Stores

ONNX

Vector Databases

Synapse/BigQuery

Dataflow/Databricks

Kafka/EventHub/PubSub

Job description

Job Description:

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 a visionary AI Engineering Lead who excels at the intersection of advanced AI expertise, strategic thinking, and impactful leadership. This role is ideal for someone who can architect and deliver enterprise-grade AI solutions, mentor AI engineering talent, and drive innovation in Generative AI. You will be instrumental in shaping our AI engineering culture, technology strategy, and customer-centric solutions, with a solid focus on accuracy, scalability, performance, and robust design.

Primary Responsibilities:

• Lead the design, development, and implementation of scalable AI/ML solutions, leveraging Python, SQL, pandas, numpy, and modern ML frameworks (TensorFlow, PyTorch, Scikit Learn)
• Lead the predictive maintenance practice for critical medical imaging equipment (MRI and CT Scanners)
• Oversee end to end ML workflows, from data preparation, feature engineering, EDA, and model development to evaluation, deployment, and continuous monitoring
• Guide model selection, optimization, and tuning, ensuring solutions balance performance, explainability, cost, and Responsible AI principles
• Build and maintain production grade ML pipelines with orchestration and experiment tracking tools such as MLflow, Feature Stores, ONNX, and vector databases
• Lead drift analysis, observability, and model health monitoring to ensure ongoing accuracy, reliability, and robustness in production environments
• Design and support scalable, multi cloud ML infrastructure (Azure/GCP primary, AWS optional), incorporating best practices for compute, storage, networking, and distributed data processing
• Collaborate with data engineering teams to integrate real time and batch pipelines built on Synapse/BigQuery, Dataflow/Databricks, and Kafka/EventHub/PubSub
• Work closely with engineering, product, data science, and UX to translate business requirements into actionable technical solutions and architecture patterns
• Review and guide the creation of architecture diagrams, C4 models, ADRs, and technical specifications as part of engineering governance
• Mentor and coach data scientists and ML engineers on best practices in model development, experimentation, coding standards, optimization, and cloud native ML deployment
• Ensure ML systems meet high standards of scalability, security, reliability, automation, observability, and operational excellence
• Collaborate with front end teams where needed to ensure ML components integrate seamlessly with product workflows and user-facing experiences
• Stay current with emerging AI/ML tools, frameworks, and research; evaluate applicability to current and future projects
• Communicate technical concepts, design trade-offs, and recommendations clearly to both technical and non technical stakeholders, including leadership
• Document methodologies, experiments, findings, and implementation details to ensure maintainability and knowledge sharing across teams
• 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:

• Bachelors or Masters degree in Computer Science, Data Science, Statistics, or a related field
• 5+ years of experience in Data Science, AI/ML, or a similar role, with 2+ years in leadership roles
• Hands‑on experience in delivering production‑grade AI/ML projects
• Experience with the full life cycle of AI/ML projects, including EDA, model development, tuning, and drift analysis
• Experience in Predictive Maintenance (using AI), IOT (Internet of Things), Sensors, Smart Manufacturing, Uptime of MRI and CT Scan machines, PHM (Prognostics and Health management), Digital Twins
• Experience with Time series models including foundational models, Signal processing, Anomaly detection, Remaining Useful life (RUL) of equipment
• Experience with streaming data, IOT architectures and tools, Edge devices and Systems integration
• Solid programming skills in Python, with experience in pandas, numpy, and SQL
• Solid understanding of mathematical and statistical concepts
• Proficiency in frameworks such as Scikit Learn, TensorFlow, and PyTorch
• Proven excellent problem‑solving and analytical thinking skills
• Demonstrated solid communication and collaboration skills to work effectively in a team environment

Preferred Qualifications:

• Knowledge of Big Data technologies (PySpark, Hadoop) and data streaming tools (Kafka, etc.)
• Familiarity with IOT and medical equipment predictive maintenance

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.

#NIC #NJP