AI or ML Engineer

Taleo; Bengaluru; On-site; INR 4,200,000 - 6,400,000; Full time

Job summary

UnitedHealth Group is seeking a visionary AI Engineering Lead to architect enterprise-grade AI solutions, mentor teams, and drive generative AI initiatives. You will shape technology strategy and ensure scalable, secure deployments across multi-cloud environments.

In this role, you will lead end-to-end ML workflows, collaborate with product and data science teams, and mentor engineers to deliver impactful health-tech solutions that improve patient outcomes and operational excellence.

Qualifications

• 5+ years of experience in Data Science, AI/ML or similar role, with 2+ years in leadership roles.
• Experience with AI/ML full lifecycle including EDA, model development, tuning, and drift analysis.
• Experience in Predictive Maintenance, IoT, Sensors, Smart Manufacturing, PHM, Digital Twins.
• Time series modeling, anomaly detection, RUL of equipment.
• Solid programming skills in Python, pandas, NumPy and SQL.

Responsibilities

• Lead design, development, and implementation of scalable AI/ML solutions using Python and modern ML frameworks.
• Oversee end-to-end ML workflows from data prep to deployment and monitoring.
• Guide model selection, optimization, and evaluation with emphasis on performance and Responsible AI.
• Mentor data scientists and ML engineers on best practices and cloud-native deployment.
• Ensure ML systems are scalable, secure, observable, and reliable.

Skills

Leadership

Communication

Problem Solving

Analytical Thinking

Education

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

Tools

Python

SQL

Pandas

NumPy

Scikit-Learn

TensorFlow

PyTorch

Job description

Improve the lives of others while Caring. Connecting. Growing together.

Job Description - AI or ML Engineer (2392825)

AI or ML Engineer - 2392825

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:

• Bachelor's or Master's 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
• 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.

UnitedHealth Group is committed to working with and providing reasonable accommodations to individuals with physical and mental disabilities. If you need special assistance or accommodation for any part of the application process, please call 1-866-566-8715 to be connected to Recruitment Services. Recruitment Services hours of operation are 7 a.m. to 7 p.m. CT, Monday through Friday.

UnitedHealth Group is a registered service mark of UnitedHealth Group, Inc. The UnitedHealth Group name with the dimensional logo, as well as the dimensional logo alone, are both service marks for the UnitedHealth Group, Inc.

Diversity creates a healthier atmosphere: UnitedHealth Group is an Equal Employment Opportunity/Affirmative Action employer and all qualified applicants will receive consideration for employment without regard to race, color, religion, sex, age, national origin, protected veteran status, disability status, sexual orientation, gender identity or expression, marital status, genetic information, or any other characteristic protected by law.

UnitedHealth Group is a drug-free workplace. Candidates are required to pass a drug test before beginning employment.