Software Engineer II: AI Compiler Engineer (AI)

TL;DR

Software Engineer II: AI Compiler Engineer (AI): Developing a deep learning compiler stack to convert neural network descriptions into optimized code for special-purpose and embedded platforms with an accent on LLVM, MLIR, and SoC performance. Focus on implementing loop optimization, IR lowering techniques, and mapping CNNs onto Xtensa processors.

Location: Hybrid (Austin, USA)

Company

hirify.global provides the expertise, tools, IP, and hardware required for the entire electronics design chain, from chip design to systems.

What you will do

• Develop a deep learning compiler stack to convert CNNs/RNNs from PyTorch, TensorFlow, and Caffe into code for embedded platforms.
• Utilize modern compiler frameworks including LLVM and MLIR.
• Implement optimized neural-network operations and integrate them into a runtime framework.
• Create optimization algorithms to map CNNs efficiently onto Xtensa processors and specialized hardware.
• Benchmark end-to-end network performance on DSP and accelerator platforms.
• Collaborate with hardware designers to identify opportunities for further hardware acceleration.

Requirements

• Bachelor's degree in Computer Science or Computer Engineering (or equivalent experience).
• 3-5+ years of high-level C and C++ programming expertise.
• Proficiency in software development on Linux and Windows systems.
• Hands-on experience with state-of-the-art compiler stacks like LLVM and MLIR.
• Knowledge of compilation techniques such as loop optimization and IR construction/lowering.

Nice to have

• Master's or PhD degree.
• 3+ years of experience working on a production compiler.
• Python programming experience.
• Experience with deep learning frameworks (TensorFlow, Caffe) and CNNs.
• Experience optimizing for embedded platforms like DSPs with DMA engines.
• Familiarity with compilation approaches like Glow, TVM, XLA, or formats like ONNX and NNEF.

Culture & Benefits

• Fortune 100 Best Companies to Work For.
• Competitive benefits package.
• Hybrid work arrangement.
• Opportunity to work on technology powering AR/VR, Vision, and intelligent IoT applications.