G

Gimlet Labs

Serverless AI inference platform with orchestration

Member of Technical Staff - GPU Performance

Software EngineeringConfirmed live in the last 24 hours
Full-timeSan Francisco, CA, USA$150k - $390k /yrIn-person

About the Job

Requirements

  • Strong software engineering fundamentals and experience working on performance-critical systems.
  • Experience profiling GPU or accelerator workloads and turning measurements into production improvements.
  • A strong understanding of GPU execution, memory hierarchies, asynchronous execution, synchronization, and performance tradeoffs.
  • Ability to design controlled experiments, distinguish symptoms from root causes, and communicate performance findings precisely.
  • Bachelor's degree in a relevant field, or an equivalent combination of education, training, and professional experience.

Responsibilities

  • Establish trustworthy performance baselines for production inference workloads across established and emerging accelerator architectures.
  • Isolate bottlenecks across compute, memory bandwidth, cache behavior, kernel launch, synchronization, host-device interaction, communication, and runtime scheduling.
  • Use profiling, hardware counters, controlled experiments, and performance models to explain workload behavior and identify the highest-leverage optimization.
  • Improve execution through launch reduction, graph capture, stream and event coordination, compute-communication overlap, memory reuse, batching, and scheduling.
  • Build automated benchmarks and regression detection that preserve performance across models, tensor shapes, data types, software releases, and hardware targets.
  • Translate performance findings into production improvements, partnering with kernel, compiler, machine learning systems, and distributed systems engineers when necessary.
  • Develop a repeatable methodology for evaluating and bringing up new accelerators within Gimlet's execution platform.

Desired Qualifications

  • Experience with Nsight Systems, Nsight Compute, rocprof, Omniperf, VTune, Perfetto, or hardware performance counters.
  • Experience with CUDA, HIP, Triton, or another accelerator programming model, and the ability to prototype low-level optimizations.
  • Familiarity with roofline analysis, queueing behavior, occupancy, latency hiding, instruction throughput, and memory-bandwidth analysis.
  • Experience optimizing inference runtimes, execution graphs, memory allocators, kernel dispatch, batching, or model-serving systems.
  • Familiarity with collective communication, topology-aware execution, and compute-communication overlap across multiple accelerators.
  • Experience building performance laboratories, benchmark suites, regression infrastructure, or hardware bring-up methodology.

About the Company

Gimlet Labs

Serverless AI inference platform with orchestration

Gimlet Labs builds an inference cloud platform that runs and manages complex AI agent workloads across heterogeneous hardware, including GPUs and other accelerators. Its software splits AI tasks and routes each part to the most suitable hardware to improve speed while keeping cost and power use the same. The Gimlet Cloud offers serverless inference for building and scaling multi-agent systems, with the platform handling scheduling and hardware-aware optimization, and it can be deployed in the cloud or in customers’ data centers. It is developing kforge, a toolkit that uses AI agents to automatically generate optimized PyTorch kernels for CUDA, ROCm, and Metal. The goal is to make AI inference a practical, scalable infrastructure problem by coordinating diverse compute to improve performance and efficiency for enterprise workloads.

Company size
51-200
Company stage
Series B
Total funding
$392M
Headquarters
San Francisco, CA
Founded
2023

Recently Posted Jobs

Explore more opportunities at Gimlet Labs

See more jobs