Staff Engineer, Inference Optimizations

DigitalOcean • San Francisco, CA

13h • $191,200 - $239,000 • Remote

About The Position

DigitalOcean is seeking a Senior Engineer 2 to play a key technical role in our AI Inference Optimization team. DigitalOcean aims to be the Inference Cloud of choice for digitally native companies and you will help ensure we can offer the industry-leading performance for our inference services. You will be responsible for the architectural decisions that maximize throughput and minimize latency for the world’s most advanced large models. As an IC leader, you will act as a force multiplier for the engineering organization, solving the most complex bottlenecks in memory bandwidth and compute utilization while guiding the technical roadmap for our high-performance inference fleet.

Requirements

• 5+ years of experience in high-performance computing or AI infrastructure, with a proven track record of solving compute utilization and memory bandwidth bottlenecks.
• Deep familiarity with the Gen AI (LLM, VLM, LMM) landscape, including the specific quirks and architectural requirements of major model families.
• Hands-on experience with attention-layer optimizations and parallelization strategies across distributed GPU environments.
• Comprehensive understanding of NVIDIA and AMD GPU architectures and their respective software ecosystems (CUDA, ROCm, etc.).
• Extensive experience integrating, building with, and contributing to open-source software projects.
• Excellent system design skills, particularly related to low-level GPU programming - optimization, memory access patterns, and parallel execution.
• Experience acting as a technical lead, driving design and delivery through cross-functional alignment and expert-level delegation.
• Deep understanding of GPU architectures (SMs, Warp scheduling, Tensor Cores).
• Expert-level Triton or CUDA. If you’ve contributed to the Triton compiler or wrote custom CUDA kernels for a major LLM, we want you.

Responsibilities

• Lead the technical strategy for benchmarking and performance optimizations at the inference engine and GPU kernel layers, ensuring our infrastructure extracts maximum value from every TFLOP.
• Engineer solutions for complex performance issues, including attention layer optimizations, memory and precision management, and advanced parallelization across multi-node GPU clusters.
• Proactively implement cutting-edge optimization techniques to keep DigitalOcean at the forefront of the Gen AI landscape. Some examples of projects you may work on: Improving batch size performance using AMD's AITER library for AMD MI355X - identify and tune AITER's CK (composable kernel) or ASK (assembly) to optimize FP8 / BF16; Identify kernel fusion opportunities for GLM-5 kernels for different layers of the Transformer block (FlashAttention, RMS Norm); Tune expert gateway router kernels for MoE models like Qwen3-235B, DeepSeek V3, GLM-5 etc
• Act as the subject matter expert on modern GPU families (NVIDIA/AMD) and their software stacks (CUDA, ROCm, TensorRT, OpenAI Triton), advising on hardware procurement and software integration.
• Develop and deploy state-of-the-art quantization techniques (FP8, INT8, and experimental FP4) to double throughput without losing accuracy.
• Lead by example through high-quality code and design reviews, elevating the technical bar for the team without the administrative overhead of direct management.
• Partner with Product Management and TPMs to translate "theoretical hardware limits" into "shippable product features," ensuring our platform is both powerful and developer-friendly.
• Maintain a strong presence in the GPU infrastructure and model performance optimization communities, contributing to and integrating the best of open-source AI.

Benefits

• Competitive array of benefits to support you from our Employee Assistance Program to Local Employee Meetups to flexible time off policy
• Reimbursement for relevant conferences, training, and education
• Access to LinkedIn Learning's 10,000+ courses
• Bonus in addition to base salary
• Equity compensation to eligible employees, including equity grants upon hire and the option to participate in our Employee Stock Purchase Program

What This Job Offers

• Job Type: Full-time
• Career Level: Senior
• Education Level: No Education Listed