Technical Lead Manager, Machine Learning Runtime & Serving
Core
Architecting and managing scalable ML runtime systems for in-vehicle inference and cloud-based serving of foundational models for autonomous driving.
Role type
Senior Technical Lead Manager (IC + Team Lead)
Builds
Next-generation ML ecosystem including in-vehicle inference engines and cloud serving infrastructure
Domain
Autonomous driving, Edge computing, Cloud infrastructure
Deliverable
production ML models
Required skills
ML system architecture, team management, hardware optimization (GPUs/TPUs), distributed backend systems, low-latency engineering, profiling & benchmarking
Preferred skills
ML compiler/runtime modification (OpenXLA/PjRT/TensorRT), LLM serving systems, edge computing deployment
Technologies
JAX, OpenXLA, PjRT, TensorRT, ONNX Runtime, TVM
Responsibilities
Guide technical vision for core ML infrastructure, architect high-performance runtime systems across edge and cloud, drive transition to JAX-native runtime, partner with researchers on system-level optimizations, design profiling infrastructure to eliminate bottlenecks
Seniority
Senior, hands-on IC with management responsibilities
