Lead ML Compiler

4 months ago

San Bruno, United States Acceler8 Talent Full time

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.

  • Lead ML Compiler

    3 weeks ago

    San Bruno, California, United States Acceler8 Talent Full time

    About the RoleWe are seeking a skilled Compiler Lead to spearhead the development of our next-generation compiler for AI 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.Key...

  • Lead ML Compiler

    3 weeks ago

    san bruno, United States Acceler8 Talent Full time

    Lead ML Compiler (MLIR)About UsWe 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...

  • ML Compiler Architect

    4 weeks ago

    San Bruno, California, United States Acceler8 Talent Full time

    About UsWe're a cutting-edge technology company that designs chips and AI algorithms for efficient edge device inference. Our focus is on distributed, near-memory-compute architecture and sparse mathematics acceleration. Our systems are integrated into various devices, including hearables, smart home appliances, and security monitors. Our mission is to...

  • Machine Learning Compiler Engineer

    3 weeks ago

    San Jose, California, United States AMD Full time

    Unlock the Power of Machine Learning with AMDAt AMD, we're pushing the boundaries of innovation to solve the world's most complex challenges. We're seeking a skilled Machine Learning Compiler Engineer to join our team of talented industry specialists working on developing a cutting-edge machine learning model compiler targeting AMD Inference Accelerator AIE...

  • Machine Learning Compiler Engineer

    3 weeks ago

    San Jose, California, United States Advanced Micro Devices , Inc. Full time

    Transforming Lives with AMD TechnologyWe're on a mission to build innovative products that accelerate next-generation computing experiences. As a skilled engineer, you'll join a core team of industry specialists working on developing a cutting-edge machine learning model compiler targeting AMD Inference Accelerator AIE hardware devices.The Role:Develop a...

  • Machine Learning Compiler Engineer

    3 weeks ago

    San Jose, California, United States AMD Full time

    About the Role:AMD is seeking a skilled Machine Learning Compiler Engineer to join a core team of talented industry specialists working on developing a cutting-edge machine learning model compiler targeting AMD Inference Accelerator AIE hardware devices.The ideal candidate will be responsible for implementing and improving effective algorithms and techniques...

  • Compiler Performance Engineer

    4 weeks ago

    San Francisco, California, United States CentML Full time

    About UsWe're a cutting-edge company, CentML, revolutionizing the AI landscape by making ML models more accessible and affordable. Our mission is to empower anyone to harness the power of AI, and we're committed to making this vision a reality.Our founding team consists of experts in AI, compilers, and ML hardware, with a proven track record of innovation at...

  • Deep Learning Compiler Engineer for Ryzen AI NPU

    3 weeks ago

    San Jose, California, United States Advanced Micro Devices , Inc. Full time

    Transforming Lives with AMD TechnologyWe are committed to enriching our industry, communities, and the world through AMD technology. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming, and embedded.The Role:We are seeking a talented...

  • Deep Learning Compiler Engineer for Ryzen AI NPU

    3 weeks ago

    San Jose, California, United States AMD Full time

    Transforming Lives with AMD TechnologyWe are committed to enriching our industry, communities, and the world through innovative AMD technology. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming, and embedded. At AMD, we push the limits...

  • Principal Machine Learning Compiler Engineer

    2 days ago

    San Jose, United States Advanced Micro Devices, Inc. Full time

    WHAT YOU DO AT AMD CHANGES EVERYTHINGWe care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our...

  • Deep Learning Compiler Engineer for Ryzen AI NPU

    4 weeks ago

    San Jose, California, United States AMD Full time

    About the RoleWe are seeking a talented Machine Learning (ML) Compiler Software Engineer to join our growing team in the AI group at AMD. As a key member of our team, you will play a crucial role in developing software tools to deploy cutting-edge ML models on AMD's XDNA Neural Processing Units (NPU). Your primary responsibility will be to design, implement,...

  • Deep Learning Compiler Engineer for Ryzen AI NPU

    3 weeks ago

    San Jose, California, United States Advanced Micro Devices , Inc. Full time

    About the RoleWe are seeking a talented Machine Learning (ML) Compiler Software Engineer to join our growing team in the AI group and play a crucial role in developing software toolsets to deploy cutting-edge ML models on AMD's XDNA Neural Processing Units (NPU).Key ResponsibilitiesDesign and develop novel algorithms for tiling and mapping quantized ML...

  • Principal Machine Learning Compiler Engineer

    2 weeks ago

    San Jose, United States AMD Full time

    WHAT YOU DO AT AMD CHANGES EVERYTHING We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences – the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our...

  • Senior Software Engineer

    4 weeks ago

    San Francisco, United States CentML Full time

    Our founding team is made up of experts in AI, compilers, and ML hardware and has led efforts at companies like Amazon, Google, Microsoft Research, Nvidia, Intel, Qualcomm, and IBM. Our co-founder and CEO, Gennady Pekhimenko, is a world-renowned expert in ML systems who holds multiple academic and industry research awards from Google, Amazon, Facebook, and...

  • Principal Machine Learning Compiler Engineer

    2 weeks ago

    San Jose, United States Advanced Micro Devices , Inc. Full time

    WHAT YOU DO AT AMD CHANGES EVERYTHING We care deeply about transforming lives with AMD technology to enrich our industry, our communities, and the world. Our mission is to build great products that accelerate next-generation computing experiences - the building blocks for the data center, artificial intelligence, PCs, gaming and embedded. Underpinning our...

  • Software Development Engineer

    3 weeks ago

    San Jose, California, United States AMD Full time

    Transforming Lives with AMD TechnologyWe're on a mission to build innovative products that accelerate next-generation computing experiences. As a skilled software development engineer, you'll join a talented team working on cutting-edge machine learning model compilers targeting AMD AIE in Ryzen AI PC and embedded FPGA devices.The RoleDevelop a compiler that...

  • Lead AI/ML Data Scientist

    3 months ago

    San Francisco, United States Unreal Gigs Full time

    Company Overview: Welcome to the forefront of artificial intelligence and machine learning innovation! Our company is dedicated to leveraging the power of data science to drive transformative change and solve complex problems across industries. We're committed to developing cutting-edge AI and ML solutions that push the boundaries of what's possible. Join us...

  • AI/ML Tech Lead

    23 hours ago

    San Francisco, United States Programmers.io Full time

    Programmers.io is currently looking for an AI/ML Tech LeadHybrid role in San Francisco, California (1-2 Days Onsite/Week)Contract to HireResponsibilities: Design, develop, and maintain a AI/ML platform that is accurate, secure, and fast Lead workstreams within a lean and agile environment from requirements gathering to creating a plan of actionable tasks for...

  • AI/ML Tech Lead

    13 hours ago

    san francisco, United States Programmers.io Full time

    Programmers.io is currently looking for an AI/ML Tech LeadHybrid role in San Francisco, California (1-2 Days Onsite/Week)Contract to HireResponsibilities: Design, develop, and maintain a AI/ML platform that is accurate, secure, and fast Lead workstreams within a lean and agile environment from requirements gathering to creating a plan of actionable tasks for...

  • AI/ML Tech Lead

    4 weeks ago

    San Francisco, California, United States Forge Global Full time

    Job Title: AI/ML Tech LeadForge Global is seeking an experienced AI/ML Tech Lead to join our team. As a key member of our engineering team, you will be responsible for leading the development of AI-powered solutions and mentoring junior engineers.Key Responsibilities:Design, develop, and maintain AI/ML platforms that are accurate, secure, and fast.Lead...