AI Compiler Engineer

Job Description

At MediaTek, we are at the forefront of enabling device+cloud hybrid architectures for generative AI. On-device AI offers significant advantages, including low latency, enhanced security, privacy, and personalization through fine-tuning to meet specific user needs. A critical component of this technology is the AI compiler, which involves code generation and optimization to maximize runtime performance and minimize memory footprint. We are seeking talented individuals to join our team and contribute to this cutting-edge field.

Key Responsibilities

• Develop and optimize AI compilers for on-device applications.
• Conduct performance analysis and implement optimizations for runtime efficiency and memory usage.
• Work closely with customers to understand their needs and provide tailored AI solutions.
  
  

Requirement

• System Software Development:
  
  

Proficiency in writing robust system software using C++.

Experience with software development best practices, including version control, testing, and debugging.

• Compiler Technology:
  
  

Strong understanding of compiler technologies.

Familiarity with ML compilers such as TVM, XLA, MLIR.

Experience with LLVM/GCC backend development, including optimization analysis and implementation, is a plus.

• Generative AI experience:
  
  

Knowledge of generative AI and its applications.

Familiarity with AI frameworks such as PyTorch, TensorFlow, and ONNX is a plus.