Staff Engineer, Inference Optimizations

Posted Yesterday
Be an Early Applicant
Boston, MA, USA
In-Office
191K-239K Annually
Senior level
Artificial Intelligence • Cloud • Software • Infrastructure as a Service (IaaS)
DigitalOcean is the Inference Cloud built for production AI.
The Role
Lead architecture and low-level optimizations for LLM inference to maximize throughput and minimize latency. Profile and optimize GPU kernels, implement quantization and parallelization strategies (FP8/INT8/FP4), tune libraries (AITER, Triton, CUDA, ROCm), and mentor engineers while translating hardware limits into shippable platform features.
Summary Generated by Built In

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. 

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.

What You’ll Do:
  • Performance Architecture: 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.
  • Deep-Dive Optimization: Engineer solutions for complex performance issues, including attention layer optimizations, memory and precision management, and advanced parallelization across multi-node GPU clusters. 
  • Technological Innovation: 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
  • Hardware & Ecosystem Mastery: 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.
  • Precision Optimization: Develop and deploy state-of-the-art quantization techniques (FP8, INT8, and experimental FP4) to double throughput without losing accuracy.
  • Technical Mentorship: Lead by example through high-quality code and design reviews, elevating the technical bar for the team without the administrative overhead of direct management.
  • Strategic Collaboration: Partner with Product Management and TPMs to translate "theoretical hardware limits" into "shippable product features," ensuring our platform is both powerful and developer-friendly.
  • Community Leadership: Maintain a strong presence in the GPU infrastructure and model performance optimization communities, contributing to and integrating the best of open-source AI.
What You’ll Bring to DigitalOcean:
  • Technical Depth: 5+ years of experience in high-performance computing or AI infrastructure, with a proven track record of solving compute utilization and memory bandwidth bottlenecks.
  • Gen AI Literacy: Deep familiarity with the Gen AI (LLM, VLM, LMM) landscape, including the specific quirks and architectural requirements of major model families.
  • Optimization Expert: Hands-on experience with attention-layer optimizations and parallelization strategies across distributed GPU environments.
  • Hardware Fluency: Comprehensive understanding of NVIDIA and AMD GPU architectures and their respective software ecosystems (CUDA, ROCm, etc.).
  • Open Source Mastery: Extensive experience integrating, building with, and contributing to open-source software projects.
  • Systems Design: Excellent system design skills, particularly related to low-level GPU programming - optimization, memory access patterns, and parallel execution.
  • Leadership through Influence: Experience acting as a technical lead, driving design and delivery through cross-functional alignment and expert-level delegation.
  • Low-Level Mastery: Deep understanding of GPU architectures (SMs, Warp scheduling, Tensor Cores).
  • The Toolkit: 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.
Compensation Range: 
  • $191,200 - $239,000

*This is a remote role

JR: 2026-7625

#LI-Remote

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.

Skills Required

  • 5+ years experience in high-performance computing or AI infrastructure (compute utilization and memory bandwidth optimization)
  • Deep familiarity with Gen AI models (LLM, VLM, LMM) and their architectural requirements
  • Hands-on experience with attention-layer optimizations and distributed GPU parallelization strategies
  • Comprehensive understanding of NVIDIA and AMD GPU architectures and software ecosystems (CUDA, ROCm)
  • Expert-level experience with Triton or CUDA (writing custom CUDA kernels or contributing to Triton)
  • Experience implementing quantization techniques (FP8, INT8, experimental FP4) without significant accuracy loss
  • Proven open-source experience: integrating, building with, and contributing to OSS projects
  • Strong low-level system design skills related to GPU programming, memory access patterns, and parallel execution
  • Demonstrated technical leadership through influence (mentoring, design and code reviews, cross-functional alignment)

What the Team is Saying

DigitalOcean Compensation & Benefits Highlights

  • Healthcare Strength Health coverage is described as market‑leading across medical, dental, vision, and mental‑health, with company‑paid life and income‑protection. This positions core healthcare and protection benefits as a standout element of the package.
  • Parental & Family Support Parental leave is characterized as above‑average, frequently highlighted alongside a phased return‑to‑work program. This indicates strong support for new parents beyond baseline policies.
  • Equity Value & Accessibility The package includes recurring equity grants and access to an Employee Stock Purchase Plan. This provides meaningful ownership opportunities alongside cash compensation.

DigitalOcean Insights

Am I A Good Fit?
beta
Get Personalized Job Insights.
Our AI-powered fit analysis compares your resume with a job listing so you know if your skills & experience align.

The Company
HQ: Broomfield, CO
1,400 Employees
Year Founded: 2012

What We Do

DigitalOcean is the Inference Cloud — a full-stack, production-ready cloud platform built to run AI applications with predictable performance, sustainable economics, and radically simpler operations at scale. We are built for teams turning AI into real products — not just training models. Our advantage is not fewer features, but fewer failure modes when operating AI at scale — combining minimal operational overhead, predictable cost efficiency, and a full-stack cloud that works as a system. Hyperscalers are broad by design. Neoclouds are infrastructure-first. DigitalOcean is inference-first — with a real cloud underneath. It combines inference-optimized compute, managed inference software, and integrated cloud capabilities that reduce operational burden for teams running real workloads. Inference is the foundation—not the boundary. Everything else builds on top of it.

Why Work With Us

At DO, we do career-defining work. We innovate with AI and build cutting-edge tech. Our rewards to match that intensity - to motivate you, recognize your impact, and give you what you need to thrive. If you have a growth mindset, like to think big and bold, and are energized by the fast-paced environment, you'll find your place here.

Gallery

Gallery
Gallery
Gallery
Gallery
Gallery

DigitalOcean Offices

Remote Workspace

Employees work remotely.

We commit to both remote work and in-person collaboration. These ways of working are dependent on specific roles and are mutually agreed upon by employees. In the US, we are mainly remote. In our APAC locations, we have a hybrid in-office approach.

Typical time on-site:
Company Office Image
HQBroomfield, CO
Company Office Image
Seattle, WA
Company Office Image
Hyderabad, Telangana
Learn more

Similar Jobs

DigitalOcean Logo DigitalOcean

Senior Engineer

Artificial Intelligence • Cloud • Software • Infrastructure as a Service (IaaS)
In-Office
Boston, MA, USA
1400 Employees
139K-174K Annually

DigitalOcean Logo DigitalOcean

Senior Security Analyst I

Artificial Intelligence • Cloud • Software • Infrastructure as a Service (IaaS)
In-Office
Boston, MA, USA
1400 Employees
141K-176K Annually

DigitalOcean Logo DigitalOcean

Product Manager

Artificial Intelligence • Cloud • Software • Infrastructure as a Service (IaaS)
In-Office
Boston, MA, USA
1400 Employees
160K-199K Annually

DigitalOcean Logo DigitalOcean

Security Engineer

Artificial Intelligence • Cloud • Software • Infrastructure as a Service (IaaS)
In-Office
Boston, MA, USA
1400 Employees
140K-175K Annually

Sign up now Access later

Create Free Account

Please log in or sign up to report this job.

Create Free Account