Lead Compiler Engineer

Neurophos • Austin, TX

21h • Onsite

About The Position

We are seeking a talented ML Compiler Engineer to join our engineering team and lead the development of our compiler. This role focuses on compiler development for our novel LLM accelerator architecture. This is one of several software stacks that seamlessly bridge high-level AI workloads with our custom hybrid optical-electronic compute hardware, enabling customers to realize game-changing performance.

Requirements

• 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

Nice To Haves

• 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

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

Benefits

• 100% coverage of base health plan premiums for you and your dependents, plus HSA contributions
• Unlimited PTO
• 401(k) matching
• Stock option opportunities
• Full suite of voluntary benefits, including Dental, Vision, Life, Hospital, Critical Illness, and Accident insurance
• Personalized Benefits

What This Job Offers

• Job Type: Full-time
• Career Level: Senior