AI Compiler Engineer

This is a fully remote position, open to applicants in California.

📋 Description

• Design, architect, and implement enhancements for AI model execution on graph compilers to boost performance, decrease latency, and optimize hardware usage.

• Collaborate closely with ML researchers, hardware engineers, and software developers to design and deploy AI models, while identifying and addressing hardware-specific challenges.

• Focus on performance enhancements for neural network models, including layer fusion, operator fusion, and graph-level transformations.

• Create compiler optimizations and passes that translate high-level AI models (for instance, from TensorFlow, PyTorch) into intermediate representations (IR).

• Execute parsing, semantic analysis, and IR generation for deep learning frameworks.

• Investigate and incorporate the latest developments in compiler design, ML model optimizations, and hardware acceleration into graph compilers.

• Provide leadership, mentorship, and technical direction to a team of engineers dedicated to graph compiler optimizations.

⛳️ Requirements

• Bachelor’s or Master’s degree in Computer Science, Electrical Engineering, or a related field (Ph.D. is preferred).

• Over 3 years of experience in compiler development, particularly with a focus on AI or ML graph compilers.

• Expertise in AI graph compiler frameworks (such as MLIR, Torch-FX).

• Strong foundation in hardware architectures (like GPUs, TPUs, ASICs) and optimization techniques including fusion, quantization, and tiling.

• Familiarity with neural network operators and code generation.

• Comprehensive understanding of intermediate representations, code parsing, and semantic analysis within compiler design.

• Proficiency in C++, Python, or other programming languages that are commonly utilized in compiler development.

• Contributions to open-source AI software frameworks and libraries are a plus.

• Proven experience in leading and mentoring engineering teams with successful project outcomes.

🏝️ Benefits

• Health insurance

• Retirement plans

• Paid time off

• Flexible work arrangements

• Professional development

• Bonuses

• Stock options

• Equipment allowances

• Wellness programs