KAI-Scheduler is an open-source CNCF project focused on delivering the best scheduling experience for AI workloads on Kubernetes. Adopted by AI frontier labs, leading enterprises, and some of the largest AI infrastructure deployments in the world, KAI helps organizations efficiently run AI at scale.

KAI is designed to support any AI infrastructure—from the latest GPU and networking technologies to future hardware generations—while maximizing performance, utilization, and scalability. As a Principal Engineer for KAI, you will drive the technical direction of the project and shape the future of AI scheduling in the Kubernetes ecosystem.

What you’ll be doing:

  • Define the technical roadmap of KAI-Scheduler, driving its architecture, APIs, extensibility, performance, and long-term direction, ensuring alignment with the evolving Kubernetes ecosystem.

  • Drive the scalability strategy of KAI for massive-scale deployments (thousands of nodes and tens of thousands of GPUs), with deep understanding of Kubernetes scaling constraints and bottlenecks.

  • Apply strong algorithmic thinking to solve complex AI workload scheduling and placement challenges, balancing performance, fairness, cluster utilization, topology constraints, and scalability.

  • Engage directly with contributors, users, and customers, incorporating feedback from real-world production deployments into the project's technical direction and serving as a visible technical leader within the open-source community.

  • Proactively collaborate with related upstream projects (schedulers, AI frameworks, cluster autoscalers, Kubernetes SIGs/WGs, etc.) and represent KAI in community and ecosystem discussions.

What we need to see:

  • B.Sc. or M.Sc. in Computer Science or a related field or equivalent experience

  • 15+ years of experience in backend software development, including system design and architecture

  • 6+ years of advanced Kubernetes development experience, including designing and implementing CRDs and controllers, with deep expertise in Kubernetes internals, networking, storage, and cluster architecture.

  • Strong algorithmic background with experience solving complex optimization and distributed systems problems.

  • Superb technical level and a proven ability to mentor other engineers through code reviews, design reviews, and technical leadership.

  • Experience in maintaining open-source projects in the Kubernetes or broader cloud-native ecosystem, with a strong understanding of healthy open-source project dynamics, contributor engagement, and community collaboration.

We are an equal-opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, color, national origin, sex, gender, gender expression, sexual orientation, age, marital status, veteran status, or disability status. We will ensure that individuals with disabilities are provided reasonable accommodation to participate in the job application or interview process, to perform essential job functions, and to receive other benefits and privileges of employment. Please contact us to request accommodation.

NVIDIA
AI & Machine Learning · 200-500 Members · Santa Clara, CA, United States

NVIDIA builds the GPUs and the CUDA software stack that most modern AI is trained and served on, along with its own research in graphics, robotics and foundation models.

All jobs at NVIDIA
Do you match this job?

Sign in and we will show how your role, experience, salary and location line up against what this employer asked for.

Check my match

Location

Israel, Tel Aviv

Job Overview
Job Posted:
2 months ago
Job Expire:
Not specified
Workplace
On-site
Job Type
Full Time
Education
Any
Experience
8+ years
Total Vacancies
1
Profession

Share This Job: