78 jobs for "Staff Software Engineer - Machine Learning"

Staff Software Engineer - Machine Learning

At General Motors - Remote / Flexible

Posted on Sep 11

View job Apply on company site

Staff Machine Learning Engineer - Central Technology, Anti-Cheat

At Riot Games - Los Angeles

Posted on Oct 12

View job Apply on company site

Senior Staff Engineer, Audio/Video Signal Processing & Machine Learning

At Dolby - Sydney

Posted on Sep 8

View job Apply on company site

Staff Machine Learning Research Engineer, App Store Search

At Apple - Seattle

Posted on Oct 6

View job Apply on company site

Staff AI & Machine Learning Engineer

At UKG - Noida

Posted on Sep 3

View job Apply on company site

Staff Machine Learning Engineer, Ads ML Efficiency

At Reddit - Remote / Flexible

Posted on Oct 1

View job Apply on company site

Staff Machine Learning Engineer

At Babylist

Posted on Oct 2

View job Apply on company site

Staff Machine Learning Engineer

At Babylist

Posted on Oct 2

View job Apply on company site

Staff Machine Learning Engineer

At BD - Irvine

Posted on Oct 26

View job Apply on company site

Staff Machine Learning Engineer

At Twilio - Remote / Flexible

Posted on Oct 26

View job Apply on company site

Staff Machine Learning Engineer - Edge AI

At Samsara Inc. - Remote / Flexible

Posted on Nov 6

View job Apply on company site

Staff Machine Learning Engineer - Edge AI

At Samsara Inc. - Remote / Flexible

Posted on Nov 6

View job Apply on company site

Staff Machine Learning Engineer

At Capital One - Plano · McLean

Posted on Oct 1

View job Apply on company site

Senior Staff Machine Learning Engineer

At Capital One - McLean · Plano

Posted on Oct 1

View job Apply on company site

Senior Staff Machine Learning Engineer, Feed Relevance

At Reddit - Remote / Flexible

Posted on Aug 11

View job Apply on company site

Senior Staff Machine Learning Engineer, Post Training

At Airbnb - Remote / Flexible

Posted on May 5

View job Apply on company site

Senior Staff Machine Learning Engineer, GenAI Platform

At Reddit - Remote / Flexible

Posted on Apr 1

View job Apply on company site

Staff Machine Learning Engineer

At BD - Irvine

Posted on Sep 30

View job Apply on company site

Staff Machine Learning Engineer

At Twilio - Remote / Flexible

Posted on Oct 23

View job Apply on company site

Sr. Staff Software Engineer

At Palo Alto Networks - Santa Clara

Posted on Oct 20

View job Apply on company site

UKG

Staff AI & Machine Learning Engineer

Noida, IndiaPosted Sep 3, 2026

Apply on company site View full job

Why UKG:

At UKG, the work you do matters. The code you ship, the decisions you make, and the care you show a customer all add up to real impact. Today, tens of millions of workers start and end their days with our workforce operating platform. Helping people get paid, grow in their careers, and shape the future of their industries. That's what we do.

We never stop learning. We never stop challenging the norm. We push for better, and we celebrate the wins along the way. Here, you'll get flexibility that's real, benefits you can count on, and a team that succeeds together. Because at UKG, your work matters-and so do you.

About the Role:

UKG is a leader in the HCM space, and is at the forefront of artificial intelligence innovation, dedicated to developing cutting-edge generative AI solutions that transform the HR / HCM industry and enhance user experiences. We are seeking a talented and motivated AI Engineers to join our dynamic team and contribute to the development of next-generation AI/GenAI based products and solutions. This role will provide you with the opportunity to work on cutting-edge SaaS technologies and impactful projects that are used by enterprises and users worldwide. As a Lead Software Engineer, you will be involved in the design, development, testing, deployment, and maintenance of software solutions. You will work in a collaborative environment, contributing to the technical foundation behind our flagship products and services.

Responsibilities

Lead architecture and delivery of cloud micro-services in Python with LangChain, LangGraph, ADK, MCP, A2A and web-first frameworks such as FastAPI, Streamlit, Gradio, FastHTML.
Own GenAI initiatives end-to-end-data strategy, model selection or fine-tuning, evaluation, deployment, and continuous optimization for structured reasoning, autonomous planning, image/video generation, and real-time live-mode interactions.
Set engineering standards for reliability, security, performance, CI/CD, and observability; enforce them through rigorous code reviews and mentoring.
Drive cross-functional execution-scope epics with product, align roadmaps, and deliver high-value features on time while proactively mitigating risk.
Champion production excellence-define SLOs, monitor telemetry, lead incident response, and run blameless root-cause analyses in a true DevOps model.
Mentor and multiply talent-coach engineers, run lightning talks, and foster a culture of candid feedback and continuous learning.
Document and broadcast decisions through clear design docs and architecture reviews that scale knowledge across the organization.

Basic Qualifications

Bachelor's degree in CS, Engineering, or equivalent practical experience (Master's preferred).
6+ years of professional Python development, including large-scale, production AI/ML systems.
Demonstrated experience in building applications on foundation-model APIs and to use AI daily in your workflow (GitHub Copilot, Windsurf/Cursor, Claude Code, etc.).
Hands-on cloud experience (GCP, AWS, or Azure) plus containerisation (Docker, Kubernetes) and modern CI/CD pipelines.
Demonstrated leadership in setting technical direction and mentoring engineers.

Preferred qualifications:

Deep expertise with LangChain, LangGraph, ADK, MCP, A2A and multimodal model orchestration.
Experience in Next-gen AI coding agents/IDEs such as Cursor, Windsurf, Claude Code.
Knowledge of GCP services (Vertex AI, BigQuery, GKE, Kubeflow), Terraform, GitHub Actions, Docker, Kubernetes.
Knowledge of Vector, SQL, and NoSQL data stores; streaming frameworks; distributed-systems concepts; advanced observability and SRE practices.
A self-starter who devours research papers and turns them into production-ready prototypes.
Energized by ambiguity and full ownership in an AI-first culture; you instinctively raise the bar for those around you.
Inclusive, candid, and relentlessly customer-obsessed.

*** Please add a link in your résumé (GitHub repo, blog post, Kaggle entry, etc.) that shows something you taught yourself in the last six months and why it excited you.***

Company Overview:

UKG is the Workforce Operating Platform that puts workforce understanding to work. With the world's largest collection of workforce insights, and people-first AI, our ability to reveal unseen ways to build trust, amplify productivity, and empower talent, is unmatched. It's this expertise that equips our customers with the intelligence to solve any challenge in any industry - because great organizations know their workforce is their competitive edge. Learn more at ukg.com.

UKG is proud to be an equal opportunity employer and is committed to promoting diversity and inclusion in the workplace, including the recruitment process.

Disability Accommodation in the Application and Interview Process

For individuals with disabilities that need additional assistance at any point in the application and interview process, please email UKGCareers@ukg.com

About UKG

Size: 10,000+ employees
Industry: Technology

View company profile