AI/ML Compiler Engineer

SAPEON: We Make AI Happen

Our mission is to create cutting-edge technology that serves humanity. 
 
As an AI-focused semiconductor chip and systems developer, we continuously explore new and innovative ways to solve real-world problems. Our team is dedicated to pushing the boundaries of what is possible and passionate about making a real impact. 
 
With our flagship series of AI semiconductor chips, we deliver unparalleled performance and efficiency in machine learning and deep learning applications across various industries, including security, media, manufacturing, mobility, and automotive.
 
Join our journey to create a happier, more enriched humanity that can address today's challenges while making strides toward a greener future.

About the role

The AI/ML Compiler Engineer is responsible for designing, developing, and maintaining compilers specifically for use in the field of artificial intelligence and machine learning. You will work closely with chip designers and software engineers to create efficient and optimized compilers that can run on AI-specific hardware.

What you'll do

  • Design and implement new compiler features for AI and machine learning applications
  • NPU compiling (backend optimization)
  • Optimize computation graph
  • Computation resource allocation and scheduling
  • DRAM, scratch pad memory management
  • Support deep learning training and inference
  • Optimize and improve existing compilers for use on AI-specific hardware
  • Optimize development of quantization, calibration, quantization-aware training, etc
  • Collaborate with chip designers and software engineers to understand and address their needs
  • Stay up-to-date with the latest advancements in compiler technology for AI and machine learning


Requirements
  • Bachelor’s in Computer Science or Computer Engineering required
  • Master’s or Ph.D. in a related field preferred
  • 5+ years of working experience in compiler development required
  • 3+ years of working experience with tensor compilers (TVM, MLIR, …) and NPU development required
  • Experience in neural network training, inference, and algorithms
  • Excellent understanding of computer architecture and optimization
  • Knowledge of AI-specific hardware is a plus
  • Strong analytical and problem-solving skills


Benefits
  • Competitive compensation + equity plan 
  • Medical/Dental/Vision/Life Insurance with no premium cost
  • Retirement plan with company matching
  • PTO and paid holidays

The pay range for this role is:

200,000 - 250,000 USD per year (SAPEON HQ)

R&D

Santa Clara, CA

Share on:

Terms of servicePrivacyCookiesPowered by Rippling