Principal MLCompiler Engineer

Lead ML Compiler (MLIR)

About Us

We design chips and AI algorithms for efficient edge device inference. Our technology focuses on distributed, near-memory-compute architecture and sparse mathematics acceleration. Our systems are integrated into hearables, smart home appliances, security monitors, and more. Our mission is to enable native AI applications on devices, overcoming existing limitations.

About the Company

We work on cutting-edge technology with best-in-class energy consumption, area, speed, and cost. Our products are embedded in designs by leading consumer electronics and chip companies, supporting applications like always-on speech enhancement, super resolution, and speech-to-text. We aim to push the boundaries of what's possible with edge AI.

About the Role

As the Compiler Lead, you will spearhead the development of our next-generation compiler for our hardware products. The compiler translates code from machine learning frameworks like PyTorch into assembly. You will own the compiler architecture, focus on the backend implementation, and grow the team to ensure production quality. Collaboration with algorithm and chip design teams will be essential to create an integrated algorithm-compiler-hardware system, optimizing the AI developer experience and workload efficiency.

What We Can Offer You

• Competitive salary and benefits
• Opportunity to work on groundbreaking technology
• Collaborative and innovative work environment
• Professional growth and development opportunities
• Flexible working hours and remote work options

Key Responsibilities

• Design and develop an ML compiler using MLIR and TVM
• Lead the compiler backend development
• Implement optimization passes
• Optimize workload fragmentation across a multi-core architecture
• Develop codegen techniques for novel instruction set features
• Maintain requirements, solution blueprints, and schedules
• Collaborate with ML and hardware teams on cross-disciplinary projects

Key Qualifications

• 5+ years in professional software development
• Experience with ML compilers, specifically MLIR
• Key contributor to previous ML compiler projects
• Proficiency in C++
• Expertise in compiler concepts (e.g., SSA form, optimization passes)
• Familiarity with distributed systems (e.g., MPI)
• Understanding of neural network layers and concepts
• MS or PhD in Computer Science, Electrical Engineering, or related field
• Excellent technical communication and collaboration skills

Desirable Skills

• Knowledge of hardware architecture concepts
• Familiarity with fixed-point quantization
• Experience with other ML compiler frameworks (e.g., TVM)
• Startup experience
• Proficiency in Python and cross-platform development (Windows, Linux)

Keywords: ML compiler, backend development, C++, MLIR, distributed systems, neural networks, hardware architecture, Python.