G

Gimlet Labs

Serverless AI inference platform with orchestration

Member of Technical Staff - Compiler Engineer

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

About the Job

Requirements

  • Experience building machine-learning compilers and runtimes with a strong focus on performance optimizations.
  • Experience with MLIR, LLVM, or comparable compiler infrastructure.
  • Experience designing intermediate representations, writing compiler passes, or implementing lowering and code-generation pipelines.
  • Strong C++ and/or Python skills.
  • A strong understanding of static single assignment, memory systems, scheduling, and hardware efficiency.
  • Experience with operator fusion, tiling, layout transformations, or memory planning.
  • Experience using profilers to investigate performance across software and hardware boundaries.
  • Experience with IREE, XLA, TVM, Triton, or similar machine-learning compilers and programming tools.

Responsibilities

  • Scale the compiler architecture to support a growing set of heterogeneous compute architectures, developing abstractions, dialects, and lowering paths that support new targets and generate code competitive with vendor-native toolchains.
  • Optimize end-to-end performance through operator fusion, target-agnostic middle-end optimizations, target-specific optimizations, and execution-planning decisions for partitioning computation across devices and moving intermediate state.
  • Evaluate and integrate agent-driven kernel generation as a code-generation strategy, determining when generated kernels outperform library or hand-written kernels and how the compiler can select them automatically and safely.
  • Support new models and serving techniques in production so new architectures and inference strategies run efficiently without per-target hand-tuning.
  • Ship compiler optimizations that measurably improve production latency, throughput, utilization, or cost.
  • Help bring a new accelerator architecture into production.
  • Reduce the engineering work required to support additional hardware targets.
  • Design execution strategies for partitioning workloads across heterogeneous hardware.
  • Extend how the compiler selects between generated, library, and hand-written kernels.
  • Enable a new model architecture or serving technique across multiple targets.

Desired Qualifications

  • Experience optimizing machine-learning inference or model-serving workloads.
  • Familiarity with kernel dispatch, launch APIs, memory allocators, or runtime interfaces.
  • Experience across GPU and non-GPU accelerator architectures.
  • Experience with automated kernel generation, autotuning, or search-based optimization.
  • Contributions to open-source compiler infrastructure.

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