Skip to content
Luma AILuma AIRedwood City, CA

Tech Lead Manager, Inference

Tech Lead Manager to hands-on lead the inference platform team at Luma AI. Own the full serving stack for multimodal models across thousands of GPUs, spending 50%+ time as IC on architecture, optimization, and debugging while growing the team and setting technical direction.

Salary not listed
Hybrid8+ YOEML Engineering

About the role

What You’ll Do

  • Spend at least half your time hands-on in the serving stack: architect and build core platform components, own the hardest design decisions, and debug the toughest production incidents yourself.
  • Lead, grow, and develop the inference engineering team: own hiring, coaching, and career growth, and build the team’s operational culture — on-call, incident response, capacity planning, and postmortems.
  • Set the technical roadmap for the serving platform: model serving engines, request routing and scheduling, autoscaling, caching, observability, and deployment pipelines.
  • Own the platform’s SLOs and economics: latency and availability targets, GPU utilization, and cost per generation across every model we serve.
  • Partner closely with research to ship new model architectures into production on day zero, and to integrate serving into online RL and evaluation loops.
  • Manage and optimize inference workloads across heterogeneous fleets — multiple clusters, clouds, and GPU vendors — including capacity planning and hardware bring-up.
  • Build sophisticated scheduling and queueing systems that optimally leverage expensive GPU resources against live traffic patterns, cluster availability, and user priority.

Representative Projects

  • Design intelligent routing and scheduling that optimizes request distribution across thousands of GPUs in multiple regions and clouds.
  • Stand up disaggregated prefill/decode serving with tiered KV-cache reuse across GPU memory, DRAM, NVMe, and network storage.
  • Autoscale and hot-swap models across the fleet to dynamically match GPU supply with live demand across production, research, and experimental workloads.
  • Take a new multimodal architecture from research checkpoint to a production deployment serving millions of users, including quantization, speculative decoding, and precision/regression validation across hardware platforms.
  • Build end-to-end tracing that follows any inference request through its full lifetime — queueing, routing, prefill, decode, and delivery.
  • Integrate the inference stack into an online reinforcement learning pipeline where serving throughput directly gates training progress.

Background

  • 8+ years of engineering experience in large-scale distributed systems or ML infrastructure, with several years building and operating model-serving or inference platforms in production.
  • Experience running inference platforms at scale — you have operated fleets on the order of thousands of GPUs across multiple clusters or clouds, and you understand what breaks at that scale.
  • Technical leadership experience, including managing or leading engineers through periods of rapid growth — and a genuine desire to keep at least half your time in hands-on technical work rather than move into pure management.
  • Deep, practical expertise in LLM and foundation-model serving engines (vLLM, SGLang, TensorRT-LLM, or equivalent) — ideally you’ve modified engine internals, debugged edge cases under load, and contributed improvements back.
  • Strong command of the serving-performance toolkit: continuous batching, KV-cache management, quantization, speculative decoding, and parallelism strategies (TP/EP/pipeline).
  • Strong Python and PyTorch; experience operating services on Kubernetes at scale.
  • Experience with queues, scheduling, traffic control, and fleet management at scale.

Bonus Points

  • Experience serving diffusion, video, or other multimodal generative models (not just text), and with FFmpeg/multimedia processing.
  • Experience with modern networking stacks — RDMA (RoCE, InfiniBand), NVLink — including KV-cache transfer and multi-node serving topologies.
  • Experience across heterogeneous accelerator platforms (NVIDIA, AMD, TPU, Trainium) and the porting/validation work that comes with them.
  • Contributions to open-source serving infrastructure (vLLM, SGLang, Ray, Kubernetes ecosystem).
  • Systems-language depth (Rust, C++, CUDA/HIP) for kernel- and runtime-level optimization.

Skills

Distributed SystemsML InfrastructureModel ServingInference PlatformsvLLMSglangTensorrt-LlmPythonPyTorchKubernetesContinuous BatchingKv Cache ManagementQuantizationSpeculative DecodingParallelism Strategies

Similar roles

ML Engineering jobs
Scale AI

Senior Machine Learning Engineer, Agent Oversight

Scale AISan Francisco, CA +1

Senior ML Engineer building observability, evaluation frameworks, and improvement loops for production agentic AI systems. Requires 5+ years production ML/LLM experience, strong grounding in agent design or evaluation, and hands-on work taking systems from prototype to scale.

216k – 270k
On-site5+ YOEML Engineering
Scale AI

Senior Software Engineer, Agent Oversight

Scale AISan Francisco, CA +1

Build platform infrastructure for observing, evaluating, and improving production AI agents at scale. Requires 4+ years software engineering experience with ML/LLM systems, strong backend/distributed systems skills, and collaboration with ML engineers.

216k – 270k
On-site5+ YOEML Engineering
Ambient.ai

Senior Software Engineer, AI Infrastructure

Ambient.aiRedwood City, CA

Build and optimize scalable AI infrastructure for real-time inference, evaluation, and continuous improvement of LLMs, LVMs, computer vision, and multimodal models on large-scale video data. Requires 4+ years production ML systems experience, strong Python skills, and expertise in inference optimization and model serving.

168k – 205k
Hybrid4+ YOEML Engineering
SpotOn

Senior Software Engineer

SpotOnChicago, IL +3

Senior AI Engineer building agentic workflows and orchestration layers to automate manual business processes for operations, sales, and customer support teams. Requires 7+ years software engineering experience, production LLM/agent expertise, and strong Python/TypeScript skills.

160k – 190k
Hybrid7+ YOEML Engineering
Snowflake

Senior AI Engineer, Enterprise

SnowflakeMenlo Park, CA

Senior AI Engineer building scalable backend services, distributed systems, and reusable AI platform components (agent frameworks, evaluation pipelines, SDKs) that power Snowflake's internal agentic applications for GTM teams. Requires 5+ years software engineering experience, strong Python/FastAPI skills, and production LLM experience.

156k – 224k
On-site5+ YOEML Engineering