Lead Compiler Engineer

Key Responsibilities

• Design and implement toolchains for our custom LLM accelerator architecture
• Develop optimization strategies that bridge software algorithms to hardware implementations
• Design and implement custom compiler components, including IR dialects, graph transformations, and lowering passes
• Optimize computational graphs and memory access patterns for our hardware architecture
• Integrate with existing ML frameworks (e.g., PyTorch, JAX, Triton).
• Build and maintain test infrastructure to ensure compiler correctness and performance

Qualifications

• Bachelor's degree in Computer Science, Computer Engineering, or related field
• 10+ years of industry experience
• 5+ years of professional experience in systems programming or compiler development
• Expert-level proficiency in Python and C
• Experience with hardware compilers
• Familiarity with Large Language Model architectures and their computational requirements
• Hands-on experience with compiler frameworks and code optimization techniques
• Deep understanding of computer architecture, memory hierarchies, and parallel computing concepts
• Experience with AI/ML accelerators (GPUs, TPUs, FPGAs) and their programming models

Preferred Skills

• Master's degree in Computer Science, Computer Engineering, or related field
• Strong background in graph theory and graph transformations in a compiler or optimization context; MLIR experience is a plus
• Experience writing programs that parse, analyze, and mutate programs as abstract syntax trees
• Experience in instrumenting and debugging parallel programs
• Experience with structured, human-supervised AI/agentic coding workflows
• Experience with LLM quantization techniques and model optimization
• Experience with high-performance computing and low-latency system design
• Familiarity with deep learning frameworks and neural network optimization

Technical Skills

• Programming Languages: Python and C (essential), Assembly
• Compiler Frameworks: LLVM, MLIR, GCC, custom backend development
• Graph Theory: Graph algorithms, graph rewriting systems, DAG optimization
• AST Processing: Parsing, analysis, and transformation of abstract syntax trees
• Testing & QA: pytest, GoogleTest, or similar frameworks; static analysis tools
• CI/CD: Jenkins, GitHub Actions, GitLab CI, or similar systems
• LLM Technologies: Transformer architectures, attention mechanisms, quantization techniques
• Development Tools: CMake, Git, Docker
• Parallel Tools: Profilers, debuggers, and instrumentation for parallel/concurrent programs

Technical Environment

What We Offer

• A pivotal role in an innovative startup redefining the future of AI hardware.
• A collaborative and intellectually stimulating work environment.
• Competitive compensation package including equity participation.
• Comprehensive benefits, including health, dental, and vision insurance.
• Opportunities for career growth and future team leadership.
• Access to cutting-edge technology and state-of-the-art facilities.
• Opportunity to publish research and contribute to the field of efficient AI inference.
• This is an opportunity to work on a game-changing technology at the intersection of photonics and AI. As part of our elite team, you’ll contribute to a platform that redefines computational performance and accelerates the future of artificial intelligence. Be a key player in bringing this transformative innovation to the world.