Firmus

Senior AI Engineer (Kubernetes & Customised Scheduler)

Singapore. Posted 2 months ago

The role

The Senior AI Engineer will design, develop, and operate a proprietary Kubernetes-native job scheduling platform for AI workloads. This role focuses on building a workload orchestration layer for massive-scale AI factories using Model-to-Grid technology.

What you will do

  • Design, build, operate, and improve the proprietary Kubernetes-native job scheduling platform.
  • Develop scheduler architecture, custom resource definitions, controllers, and APIs.
  • Implement scheduling policies for training, fine-tuning, inference, and other workloads.
  • Build topology-aware and resource-aware placement mechanisms.
  • Implement workload-control capabilities like admission control, queueing, and preemption.
  • Develop workload demand-shaping and grid-aware controls.
  • Integrate the scheduler with Kubernetes control-plane and various platform services.
  • Build developer and user experience tools including APIs, SDKs, and CLI workflows.
  • Enable Model-to-Grid benchmarking by exposing scheduler and workload data.
  • Build observability into the scheduling platform.

Required qualifications

  • 5+ years of experience in DevOps, site reliability engineering, platform engineering, distributed systems, cloud infrastructure, or comparable software and systems-engineering roles.
  • Deep practical expertise in Kubernetes architecture and operations.
  • Demonstrated experience building, extending, or deeply integrating workload schedulers, resource managers, or orchestration systems.
  • Strong programming skills in Go.
  • Python proficiency for automation, integration, tooling, and performance analysis.
  • Experience building production Kubernetes controllers, operators, scheduling plugins, admission webhooks, REST or gRPC APIs, and CLI tools.
  • Strong understanding of GPU-accelerated AI/ML workloads.
  • Practical experience with GPU scheduling and resource allocation.
  • Understanding of GPU system topology and high-performance AI networking.
  • Familiarity with large-scale NVIDIA GPU architectures.
  • Experience working with storage and data-path considerations for AI workloads.
  • Understanding of AI-factory or data-center operations.

Nice to have

  • Experience with Slurm, Slinky, vCluster, Kueue, KAI, Volcano, YuniKorn, or related systems.

About the company

Firmus Technologies is a global leader pioneering the development and operation of efficient AI infrastructure across Asia Pacific. They design, build, and operate the AI Factory using a model-to-grid technology approach.

Senior AI Engineer (Kubernetes & Customised Scheduler) at Firmus | JobZoey