Staff Engineer

Dive in and do the best work of your career at DigitalOcean. Journey alongside a strong community of top talent who are relentless in their drive to build the simplest scalable cloud. If you have a growth mindset, naturally like to think big and bold, and are energized by the fast-paced environment of a true industry disruptor, you’ll find your place here. We value winning together—while learning, having fun, and making a profound difference for the dreamers and builders in the world.

We are seeking a Staff AI Orchestration Engineer to lead the design, optimization, and scaling of our Kubernetes-based AI infrastructure. In this role, you will tackle the unique challenges of massive-scale AI workloads, focusing on throughput, GPU utilization, and fault tolerance to support next-generation distributed training and disaggregated inference.

What You'll Do

• Architect Large-Scale Scheduling: Design and optimize hierarchical, high-throughput scheduling architectures for massive Kubernetes clusters (1,000 nodes, 10,000 pods), utilizing techniques like optimistic concurrency, multi-scheduler architectures, and batch dispatching.
• Maximize GPU Utilization: Eliminate GPU waste in multi-tenant environments by implementing fractional GPU allocation, leveraging mechanisms like KAI-Scheduler's Reservation Pods or hard-isolation tools like HAMi, and configuring time-based fairshare scheduling to balance over-quota pool access.
• Optimize Placement & Topology: Deploy topology-aware scheduling to align pod placement with physical hardware dimensions, such as NVLink connections, PCIe lanes, and NUMA nodes, minimizing communication latency for multi-GPU operations.
• Enhance Cluster Performance: Reduce scheduling latency and API server load by tuning etcd, optimizing admission webhooks, and implementing in-place pod resizing (VPA) or in-place container restarts.
• Secure AI Workloads: Design secure, multi-layered isolation environments and Agent Sandboxes to safely execute untrusted LLM-generated code, utilizing namespaces, Kata Containers, gVisor, or Firecracker microVMs.
• Manage AI Storage & Fault Tolerance: Orchestrate efficient model weight distribution using OCI Image Volumes and implement Checkpoint/Restore capabilities (via CRIU and NVIDIA cuda-checkpoint) for long-running training fault recovery.
• Enable Distributed Training: Implement robust gang scheduling to prevent deadlocks in tightly-coupled, multi-node training jobs (e.g., MPI, PyTorch) using tools like Volcano, Kueue, or LeaderWorkerSet (LWS).
• Orchestrate Complex Inference: Implement and manage disaggregated AI inference pipelines using frameworks like NVIDIA Grove, coordinating multicomponent deployments (e.g., prefill leaders, decode workers, KV routers) with multilevel autoscaling and explicit startup ordering.
  
  

What You'll Bring

• Kubernetes Expertise: Deep technical knowledge of Kubernetes core components, API performance optimization, Dynamic Resource Allocation (DRA), and the custom resource definitions (CRDs) required for advanced scheduling.
• Advanced Scheduling Experience: Proven track record working with AI-specific Kubernetes schedulers and orchestrators such as Kueue, Volcano, Apache YuniKorn, or Run:ai / KAI-Scheduler.
• Hardware & Topology Acumen: Deep understanding of GPU architectures (NVIDIA and AMD) and interconnects, understanding how hardware topology directly impacts training and inference speeds.
• Resource Management Skills: Experience balancing performance and cost using Dominant Resource Fairness (DRF), load-aware scheduling, and bin-packing vs. spread strategies to maximize node vacancy or workload resources.
• Systems Isolation Background: Familiarity with container runtime internals (containerd, runc), rootless containers, and security contexts to manage blast radiuses in shared AI infrastructure.
• AI/ML Framework Knowledge: Strong understanding of modern LLM serving architectures, prefill-decode disaggregation, and engines like vLLM, Triton, or SGLang.
• Observability Proficiency: Experience tracking deep infrastructure and inference metrics, including Time To First Token (TTFT), Time Per Output Token (TPOT), GPU memory pressure, and identifying hardware failures like XID errors.
  
  

Compensation Range

• $191,200.00 - $239,000.00
• This is a hybrid role
  
  

JR: 2026-7729

Why You’ll Like Working for DigitalOcean

• We innovate with purpose. You’ll be a part of a cutting-edge technology company with an upward trajectory, who are proud to simplify cloud and AI so builders can spend more time creating software that changes the world. As a member of the team, you will be a Shark who thinks big, bold, and scrappy, like an owner with a bias for action and a powerful sense of responsibility for customers, products, employees, and decisions.
• We prioritize career development. At DO, you’ll do the best work of your career. You will work with some of the smartest and most interesting people in the industry. We are a high-performance organization that will always challenge you to think big. Our organizational development team will provide you with resources to ensure you keep growing. We provide employees with reimbursement for relevant conferences, training, and education. All employees have access to LinkedIn Learning's 10,000 courses to support their continued growth and development.
• We care about your well-being. Regardless of your location, we will provide you with a competitive array of benefits to support you from our Employee Assistance Program to Local Employee Meetups to flexible time off policy, to name a few. While the philosophy around our benefits is the same worldwide, specific benefits may vary based on local regulations and preferences.
• We reward our employees. The salary range for this position is based on market data, relevant years of experience, and skills. You may qualify for a bonus in addition to base salary; bonus amounts are determined based on company and individual performance. We also provide equity compensation to eligible employees, including equity grants upon hire and the option to participate in our Employee Stock Purchase Program.
• DigitalOcean is an equal-opportunity employer. We do not discriminate on the basis of race, religion, color, ancestry, national origin, caste, sex, sexual orientation, gender, gender identity or expression, age, disability, medical condition, pregnancy, genetic makeup, marital status, or military service.
  
  

Application Limit: You may apply to a maximum of 3 positions within any 180-day period. This policy promotes better role-candidate matching and encourages thoughtful applications where your qualifications align most strongly.