G

Gimlet Labs

Serverless AI inference platform with orchestration

Member of Technical Staff - ML Systems & Inference

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

About the Job

Requirements

  • Strong software engineering fundamentals.
  • Experience building or operating machine-learning inference or model-serving systems.
  • Comfort reasoning about performance, memory usage, and system behavior under load.
  • A bachelor's degree in a relevant field, or an equivalent combination of education, training, and professional experience.

Responsibilities

  • Build inference systems that execute models end-to-end in production.
  • Determine how inference executes across the pipeline, including how requests are batched and scheduled, how stages are placed and scaled, how key-value cache and intermediate state move between accelerators, and how the system balances latency, throughput, and utilization across different hardware characteristics.
  • Work across model serving, batching, scheduling, concurrency, key-value cache management, and memory placement.
  • Bring up models on novel hardware.
  • Support new model architectures and inference techniques, improve performance under real production workloads, and partner with compiler, kernel, networking, and distributed systems engineers to optimize the full execution path.
  • Improve the latency, throughput, and efficiency of production inference workloads.
  • Design execution strategies across batching, scheduling, concurrency, and resource utilization.
  • Improve key-value cache management, memory efficiency, and execution under load.
  • Enable new models, accelerator architectures, and inference techniques to run efficiently in production.

Desired Qualifications

  • Experience with inference runtimes such as TensorRT-LLM, vLLM, or custom serving systems.
  • Deep understanding of modern model architectures and attention mechanisms.
  • Experience with batching, scheduling, and concurrency control in inference systems.
  • Familiarity with key-value cache management and memory placement strategies.
  • Experience profiling and tuning latency- and throughput-critical systems.
  • Software development experience in Python and C++.

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