Member of Technical Staff - Compiler Engineer

Gimlet Labs, San Francisco, CA

The role

As a Member of Technical Staff, Compiler, you will build Gimlet's AI Compiler Stack: the layer that determines how AI workloads are represented, optimized, and executed across hardware with fundamentally different architectures, memory systems, and performance characteristics.

The compiler is one of the hardest and most critical parts of our inference stack. Its decisions reach directly into scheduling, communication, memory movement, kernel execution, and end-to-end serving performance of Gimlet’s inference stack. A compiler built for a single target can bake in assumptions about the memory hierarchy and execution model. We can't. That constraint is what makes the work interesting.

Our stack is built on open-source MLIR and LLVM compiler infrastructure. We design our own dialects, passes, and lowering paths on top of that infrastructure rather than maintaining a proprietary compiler from scratch. You will be working in an ecosystem you can carry with you, alongside people who know it deeply.

All jobs

Member of Technical Staff - Compiler Engineer

Gimlet Labs, San Francisco, CA

The role

What you will do

Scaling the compiler architecture to a growing set of heterogeneous compute architectures. New CPUs, GPUs, and accelerators arrive with different memory models, execution models, and toolchains, and each one can't cost us a rewrite. You will work on the abstractions, dialects, and lowering paths that let the stack absorb a new target quickly and still generate code competitive with a vendor-native toolchain. Our architecture for running AI workloads across heterogeneous hardware https://gimletlabs.ai/blog/heterogeneous-ai-infrastructure describes where the compiler sits in the broader system.
Optimizing end-to-end performance that includes implementing operator fusion, target-agnostic middle-end optimizations, target-specific optimizations and the execution-planning decisions that shape how computation is partitioned across devices and how intermediate state moves between them. This also includes evaluating and integrating agent-driven kernel generation as a first-class code generation strategy: deciding when a generated kernel beats a library or hand-written one, and how the compiler makes that call automatically and safely. We have published early results on AI-generated CUDA kernels https://gimletlabs.ai/blog/ai-generated-cuda-kernels and AI-generated Metal kernels https://gimletlabs.ai/blog/ai-generated-metal-kernels; folding this into the compiler itself is open ground.
Supporting new models and serving techniques in production. New architectures and inference strategies should run efficiently without per-target hand-tuning. Our work on low-latency speculative decoding https://gimletlabs.ai/blog/low-latency-spec-decode-corsair and prefill/decode disaggregation across accelerators https://gimletlabs.ai/blog/multivendor-prefill-decode-disaggregation shows the shape of these problems end to end.

Who you will work with

This is a small team where the boundaries between disciplines are thin. You will be part of the compiler team and will work directly with engineers and researchers working on LLM architecture, inference optimization, kernel engineering, distributed systems, and datacenter infrastructure deciding what gets served and on what hardware. Compiler decisions here are made with full visibility into the workload and the machine, which is rarer than it sounds.

What success looks like

In your first 12–18 months, you will

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 Gimlet selects between generated, library, and hand-written kernels.
Enable a new model architecture or serving technique across multiple targets.

Member of Technical Staff - Compiler Engineer

The role

Member of Technical Staff - Compiler Engineer

The role

What you will do

In your first 12–18 months, you will

What you need

Strong candidates may also have

About Gimlet Labs

Similar jobs at other companies

Member of Technical Staff, AI-Driven Compilation

Member of Technical Staff, GPU Compiler

Member of Technical Staff, AI-Driven Compilation

Member of Technical Staff, GPU Compiler

Compiler Code Gen Engineer

Sr. Compiler Engineer

Gimlet Labs