Job Summary:
Luminal is focused on optimizing AI models to accelerate and simplify model deployment. They are seeking a Senior Compiler Engineer to design and build core compiler infrastructure, develop backend code generation, and improve model performance across various hardware platforms.
Responsibilities:
• Design and build core compiler infrastructure in Rust
• Build search-based optimization systems for discovering faster kernels
• Develop backend code generation for multiple targets (NVIDIA, Trainium, AMD, TPU, etc.)
• Implement compiler passes for fusion, scheduling, memory planning, and kernel selection
• Profile real models, identify bottlenecks, and improve latency and throughput
• Help shape Luminal’s engineering culture from the ground up
Qualifications:
Required:
• Experience with compiler frameworks: LLVM, MLIR, or other custom compilers
• Hands-on experience with at least one of: PTX/SASS, GCN/RDNA assembly, or other GPU ISAs
• Familiar with ML compilers: torch.compile (or custom PyTorch backend), XLA, TVM, etc.
Preferred:
• Background in distributed systems or multi-device compilation
• Experience with Egg/Egglog is a STRONG plus
• Experience with Rust
• Experience in developing and deploying AI models in production environments
Company:
Luminal provides a machine learning compiler and serverless cloud platform that automates PyTorch model optimization and deployment. Founded in 2025, the company is headquartered in San Francisco, USA, with a team of 11-50 employees. The company is currently Early Stage.