Build inference systems for AI models in production at Gimlet Labs, focusing on performance and efficiency.
Posted by employer 6 months ago
First seen on Joblaze 5 hours ago
Last verified on the company career page 5 hours ago
Skills & Technologies
What you'll build
Must have
Nice to have
Requirements
Not disclosed in this posting: compensation, years of experience, work arrangement, visa sponsorship.
Joblaze summary
In this role, the Member of Technical Staff will focus on developing and optimizing machine learning inference systems for production environments. Key responsibilities include managing request batching, scheduling, and memory placement while ensuring efficient execution across various hardware setups. This position is well-suited for candidates with a strong software engineering background and experience in ML inference systems, particularly those familiar with performance tuning and modern model architectures. Gimlet Labs is currently expanding its technology offerings, providing a unique environment for tackling complex challenges.
Joblaze insights
Quick facts
From the original posting
About the role
As a Member of Technical Staff focused on ML Systems, you will build the inference systems that execute models end-to-end in production.
You will work on the systems that determine how inference executes across that pipeline: how requests are batched and scheduled, how stages are placed and scaled, how KV cache and intermediate state move between accelerators, and how the system balances latency, throughput, and utilization across different hardware characteristics.
You will work across model serving, batching, scheduling, concurrency, KV cache management, and memory placement. You will help bring up models on novel hardware. You will 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.
What success looks like
In the first 12-18 months, you will:
Improve the latency, throughput, and efficiency of production inference workloads
Design execution strategies across batching, scheduling, concurrency, and resource utilization
Improve KV cache management, memory efficiency, and execution under load
Enable new models, accelerator architectures, and inference techniques to run efficiently in production
Strong software engineering fundamentals
Experience building or operating ML inference or model serving systems
Comfort reasoning about performance, memory usage, and system behavior under load
Bachelor's degree in a relevant field, or an equivalent combination of education, training, and professional experience.
Strong candidates may also have
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 KV cache management and memory placement strategies
Experience profiling and tuning latency- and throughput-critical systems
Software development experience in Python and C++
Solve hard problems.
Own meaningful work.
Build for production.
Help define what’s next.
Standard company text repeated across Gimlet Labs's postings is omitted here.