Lead ML Compiler

5 days ago


San Bruno, California, United States Acceler8 Talent Full time

About the Role

We 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 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

What We Offer

  • 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

Requirements

  • 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)

  • ML Compiler Lead

    3 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 innovative approach focuses on distributed, near-memory-compute architecture and sparse mathematics acceleration. Our systems are integrated into various devices, including hearables, smart home appliances, security monitors, and...

  • ML Compiler Architect

    2 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...


  • 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...


  • 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...


  • 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...


  • San Jose, California, United States SilicaRise Full time

    Job DescriptionWe are seeking a highly skilled Senior Compiler Engineer to join our team at SilicaRise.Responsibilities:Design and implement compilers for DSA, ensuring optimal performance and efficiency.Conduct performance analysis and optimization for ML workloads and kernels, driving innovation in compiler technology.Collaborate with cross-functional...


  • 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...


  • 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...


  • 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...


  • 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,...


  • 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...


  • San Jose, California, United States AMD Full time

    Transforming Lives with AMD TechnologyWe are passionate about using technology to enrich our industry, communities, and the world. Our mission is to build innovative products that accelerate next-generation computing experiences, from the data center to PCs, gaming, and embedded systems. At AMD, we push the limits of innovation to solve the world's most...


  • 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...


  • San Francisco, California, United States Unreal Gigs Full time

    Job Title: Lead AI/ML Data ScientistCompany Overview:Welcome to Unreal Gigs, a pioneering company at the forefront of artificial intelligence and machine learning innovation. Our mission is to harness 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...


  • San Francisco, California, United States Strativ Group Full time

    Senior ML Engineering LeadCompensation: up to $450k base + ~1-1.5% equityWe are collaborating with a well-funded startup that is transforming the domain of language models (LLMs) through advanced intelligence and optimization techniques, allowing both businesses and individuals to leverage LLMs tailored to their specific requirements.This organization has...

  • AI/ML Tech Lead

    2 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...


  • San Diego, California, United States BioSpace, Inc. Full time

    Job SummaryWe are seeking a highly skilled Tech Lead to drive the development and execution of innovative AI/ML-based solutions that advance drug discovery and development at Eli Lilly. As a key member of our team, you will work closely with the Product Owner to co-define delivery milestones for the AI/ML platform and lead the design, development, and...


  • San Diego, California, United States BioSpace, Inc. Full time

    Job DetailsLilly is a global healthcare leader that unites caring with discovery to make life better for people around the world. Our employees work together to discover and bring life-changing medicines to those who need them, improve the understanding and management of disease, and give back to our communities through philanthropy and volunteerism.What...


  • San Francisco, California, United States Forge Global Full time

    Job Title: Tech Lead, AI/MLAt Forge Global, we're pushing the boundaries of innovation in the private market. As a Tech Lead in AI/ML, you'll play a pivotal role in developing and mentoring our data science team, driving the creation of AI-powered products that transform our business.Key Responsibilities:Design, develop, and maintain a robust AI/ML platform...


  • San Jose, California, United States AMD Full time

    Transforming Lives with AMD TechnologyWe are seeking a highly skilled AI/ML software architect to join our team at AMD. As a key member of our AI group, you will be responsible for architecting runtime stacks, defining operator mapping and dataflow, and scheduling operators on AMD's XDNA Neural Processing Units.Your work will directly impact the efficiency,...