AI or ML Engineer

Posted 3 Hours Ago
Be an Early Applicant
Bengaluru, Bengaluru Urban, Karnataka, IND
In-Office
Mid level
Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
The Role
Design, implement, and operate production-ready AI/ML solutions including generative AI and LLM-based systems. Build large-scale training and inference pipelines on Databricks and cloud platforms, deploy services via REST/FastAPI and Docker, apply RAG/LangChain and vector DBs, optimize and quantize models, and mentor engineers while ensuring governance, CI/CD, and MLOps practices.
Summary Generated by Built In
Requisition Number: 2365630
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 AI/ML Engineer with solid software engineering fundamentals and growing depth in AI/ML and data platforms. This role emphasizes building production ready, scalable AI services, applying Generative AI techniques, and continuously expanding expertise across machine learning and data engineering domains.
Primary Responsibilities:
  • AI/ML Solution Development
    • Design, implement, and optimize AI and machine learning solutions, including statistical models, deep learning, and Generative AI systems
  • Model Training, Evaluation & Optimization
    • Execute proof of concepts, train models at scale, and baseline performance using quantitative evaluation metrics
  • Platform & Infrastructure Engineering
    • Build and operate large scale training and inference pipelines using Databricks, PySpark, and cloud platforms (AWS, Azure, GCP)
  • Generative AI & Advanced Techniques
    • Apply RAG, LangChain, and Vector Databases to develop GenAI solutions
    • Optimize and quantize models to improve performance, scalability, and cost efficiency
  • Software Engineering & APIs
    • Develop REST and FastAPI services, containerize solutions using Docker, and integrate UI tools such as Streamlit or Flask
  • Collaboration & Communication
    • Partner with cross functional teams to translate business needs into clear, scalable AI solutions, and present insights effectively
  • Leadership, Mentorship & Culture
    • Mentor engineers, participate in design and architecture reviews, and uphold standards for quality, safety, and trust
  • 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 regards 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

Candidate Profile & Growth Mindset
  • The most solid alignment is with software engineering fundamentals, including system design, clean and maintainable coding, and building production ready services end to end, from development through deployment and cloud infrastructure
  • Comfortable working across AWS and cloud native architectures, with increasing hands on application of AI concepts, particularly LLMs and RAG based solutions
  • Completed AI Dojo Generative AI training, strengthening applied Generative AI foundations and practical implementation skills
  • Actively developing deeper expertise in core machine learning and data engineering, including:
    • Traditional ML models, algorithms, and evaluation techniques
    • Feature engineering and data pipelines
    • Large scale data processing using Spark and Databricks
  • Highly motivated to close skill gaps through structured training, mentorship, and hands on learning, and values collaboration and knowledge sharing within the team

Required Qualifications:
  • Bachelor's or Master's degree in Computer Science, Data Science, or a related field
  • Software & AI/ML Engineering Experience
    • 3+ years of professional software engineering experience, delivering high quality, production grade commercial applications end to end
    • 1+ years of AI/ML engineering experience, including deploying models at scale and contributing to technical leadership across AI initiatives
    • Demonstrated ability to design, build, deploy, and operate production ready services, including CI/CD and cloud infrastructure
  • Programming & Systems Expertise
    • 2+ years of hands on experience with Java, Python, SQL, and scripting
    • Solid foundation in clean, maintainable code, system design, API development, and modern software engineering best practices
  • Cloud & Data Platform Experience
    • 1+ years of experience across AWS, Azure, and GCP, with deeper hands on experience in AWS and cloud native architectures
    • 1+ years of experience with Databricks, MongoDB, PySpark/SparkSQL, and data pipeline implementation
    • Familiarity with Hadoop ecosystems and distributed data processing
  • MLOps, Infrastructure & Governance
    • Experience with data governance concepts, including access control and platform level controls in Databricks (Delta Lake, Unity Catalog)
    • Working knowledge of MLOps practices, including model lifecycle management and operationalization
    • Familiarity with Infrastructure as Code using Terraform and CloudFormation
  • Technical Expertise
    • Solid knowledge of AI/ML frameworks, orchestration tools, and scalable architectures
    • Proficiency with big data technologies, including Spark, Hadoop, and Kafka
    • Solid foundation in data science principles, including statistics, probability theory, optimization, simulation, and data modeling

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.

Skills Required

  • Bachelor's or Master's degree in Computer Science, Data Science, or related field
  • 3+ years professional software engineering experience delivering production-grade applications
  • 1+ years AI/ML engineering experience, including deploying models at scale
  • 2+ years hands-on experience with Java, Python, SQL, and scripting
  • 1+ years experience with Databricks, PySpark/SparkSQL, and MongoDB
  • Experience across AWS, Azure, and GCP (deeper hands-on experience in AWS)
  • Familiarity with Hadoop ecosystem and distributed data processing
  • Experience with data governance concepts and Databricks platform controls (Delta Lake, Unity Catalog)
  • Working knowledge of MLOps practices, model lifecycle management, and operationalization
  • Familiarity with Infrastructure as Code (Terraform, CloudFormation)
  • Proficiency with big data technologies including Spark and Kafka
  • Demonstrated ability to design, build, deploy, and operate production-ready services including CI/CD and cloud infrastructure
  • Experience developing REST APIs and FastAPI services, containerization with Docker, and integrating UI tools (Streamlit or Flask)
  • Experience applying Generative AI techniques (LLMs, RAG), LangChain, and vector databases; model optimization and quantization
  • Solid foundation in statistics, probability, optimization, simulation, and data modeling
  • Completed AI Dojo Generative AI training

What the Team is Saying

Optum Compensation & Benefits Highlights

  • Healthcare Strength Health coverage offers copay and HSA medical options with dental, vision, company‑paid life and disability, and free or low‑cost virtual visits. Feedback suggests the offering is comprehensive and competitive on paper.
  • Parental & Family Support Time off and family supports include PTO, eight paid holidays plus a floating day, six weeks paid parental leave, up to two weeks paid caregiver leave, Bright Horizons back‑up care, and adoption assistance up to $10,000. Feedback suggests these resources are meaningful for caregivers and family needs.
  • Retirement Support Savings programs include a 401(k) with employer match (after one year, vesting after two) and a 10%‑discount Employee Stock Purchase Plan. These programs bolster long‑term financial security when combined with other savings resources.

Optum Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Eden Prairie, MN
160,000 Employees
Year Founded: 2011

What We Do

Optum, part of the UnitedHealth Group family of businesses, 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. At Optum, we support your well-being with an understanding team, extensive benefits and rewarding opportunities. By joining us, you’ll have the resources to drive system transformation while we help you take care of your future. We recognize the power of connection to drive change, improve efficiency and make a difference in health care. Join a team where your skills and ideas can make an impact and where collaboration is key to creating technology that produces healthier outcomes.

Gallery

Gallery
Gallery
Gallery

Optum Offices

Hybrid Workspace

Employees engage in a combination of remote and on-site work.

Optum has three workplace models that balance the needs of the business and the responsibilities of each role. These models, core on‑site (5 days/week), hybrid (4 days/week) and telecommute or fully remote, vary by country, role and location.

Typical time on-site: Not Specified
HQEden Prairie, MN
Metro Manila, Philippines
Cebu, Philippines
Davao, Philippines
Ann Arbor, MI
Atlanta, GA
Baltimore, MD
Bengaluru, India
Chennai, India
Dallas, TX
Detroit, MI
Dublin, Ireland
Hartford, CT
Houston, TX
Hyderabad, India
Jacksonville, FL
Las Vegas, NV
Letterkenny, Ireland
Louisville, KY
Madison, WI
Minneapolis, MN
Nashville, TN
New Delhi, India
Philadelphia, PA
Phoenix, AZ
Pune, India
Raleigh, NC
San Diego, CA
Washington, DC
Learn more

Similar Jobs

Optum Logo Optum

Machine Learning Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Bengaluru, Bengaluru Urban, Karnataka, IND
160000 Employees

Optum Logo Optum

Machine Learning Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Bengaluru, Bengaluru Urban, Karnataka, IND
160000 Employees

Optum Logo Optum

Machine Learning Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Bengaluru, Bengaluru Urban, Karnataka, IND
160000 Employees

Optum Logo Optum

Machine Learning Engineer

Artificial Intelligence • Big Data • Healthtech • Information Technology • Machine Learning • Software • Analytics
In-Office
Bengaluru, Bengaluru Urban, Karnataka, IND
160000 Employees

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account