Sign up to access all features of our service.
  • Job search
  • Favorites
  • Create a CV
    New
  • Salaries
  • Subscriptions

Staff ML Compiler Engineer - TPU/LLVM & MLIR

Google

Google is seeking a Software Engineer in Mountain View, California, focused on compiler development for next-generation technologies. You will work on LLVM infrastructure to enhance High-Performance Computing capabilities on Google's TPU accelerators and contribute to research impacting billions of users. Ideal candidates will have strong software development skills, experience with compilers, and a deep understanding of MLIR and TPU systems. This role offers a competitive salary package and opportunities for continuous professional growth. #J-18808-Ljbffr Google

Vacancy posted 3 days ago
Similar jobs that could be interesting for youBased on the Staff ML Compiler Engineer - TPU/LLVM & MLIR in Mountain View, CA vacancy
  • $128.7k - $261.3k

     ...Mission For the AI Kernels & Compilers team, that mission shows up in...  ...development, and performance engineering so that every cycle on our accelerators...  ...reliable, and effortless for ML engineers across the AV...  ...and software stack (e.g., ONNX, MLIR, XLA, TVM, TensorRT).... 
    Suggested
    Local area
    Flexible hours

    General Motors

    Sunnyvale, CA
    3 days ago
  •  ...unique intersection of research, software engineering, hardware engineering, and product...  ...compression and acceleration, building ML compilers and runtimes for efficient execution, and...  ...) Experience with any compiler stack (MLIR/LLVM/TVM/...). Experience with any ML authoring... 
    Suggested

    Apple

    Cupertino, CA
    3 days ago
  • $207k - $300k

    Google is seeking a skilled software engineer to join the EdgeTPU compiler team in Mountain View, CA. This role involves building ML compilers for EdgeTPU hardware, optimizing compiler performance, and collaborating with other teams to enhance user experiences with machine... 
    Suggested

    Google

    Mountain View, CA
    3 days ago
  • Neurophos is hiring an ML Compiler Engineer to lead compiler development for a novel LLM accelerator. You will bridge high‑level AI workloads with hybrid optical‑electronic compute hardware, delivering game‑changing performance. Location: San Jose, CA or Austin, TX. This... 
    Suggested
    Full time

    Neurophos

    Sunnyvale, CA
    3 days ago
  • General Motors LLC in Sunnyvale, California is looking for a Staff Compiler Engineer to join the AI Kernels & Compilers team. In this role, you will own the end-to-end compilation stack that turns high-level models into optimized inference artifacts for autonomous driving... 
    Suggested

    General Motors

    Sunnyvale, CA
    3 days ago
  • Google Inc. in Sunnyvale, CA, is looking for a Senior Staff Software Engineer focused on TPU Performance. This role is crucial for developing benchmarks and optimizing ML performance for Google's Cloud services. Candidates should have extensive experience in software development... 

    Google Inc.

    Sunnyvale, CA
    19 hours ago
  • $207k - $300k

    Staff SWE, Compiler Architect, System Performance Modeling In accordance with...  ...silicon stage. Experience with ML system architectures,...  ...-source compiler frameworks (MLIR, XLA) to solve system level analysis...  ..., and talent. AI/ML software engineers in Cloud bridge the gap... 
    Full time
    Temporary work
    Worldwide

    Google

    Sunnyvale, CA
    3 days ago
  • $152k - $241.5k

     ...Senior Machine Learning Applications and Compiler Engineer! NVIDIA is seeking engineers to...  ...related spatial accelerators at top tier ML, compiler, and computer architecture venues...  ..., or code generation. Experience with LLVM and/or MLIR, including building custom passes,... 

    2100 NVIDIA USA

    Santa Clara, CA
    1 day ago
  • NVIDIA is seeking a Senior Machine Learning Applications and Compiler Engineer to craft high-performance runtime and compiler components for large-scale inference workloads. You will map neural network graphs onto NVIDIA platforms, extending the CUDA ecosystem with efficient... 

    2100 NVIDIA USA

    Santa Clara, CA
    1 day ago
  • $251k - $310k

     ...collaborative group of machine learning (ML) engineers, software engineers, and ML...  ...you will report to a Senior Staff Engineering Manager. You...  ...(TPUs, GPUs), leveraging compiler features and low-level...  ...practices. We prefer: Knowledge of TPU and GPU architectures and how... 

    Waymo

    Mountain View, CA
    3 days ago
  • A leading spatial computing firm is seeking a Staff Machine Learning R&D Engineer to innovate their platform. This role involves developing machine learning...  ...vision. Candidates should have a strong background in ML research and production, focusing on user experience and... 

    Matterport

    Sunnyvale, CA
    3 days ago
  • Google Inc. is looking for a Senior Staff Software Engineer in Sunnyvale, California. In this role, you will focus on optimizing machine learning...  ..., particularly with Python and C++, and an understanding of ML infrastructure. The role offers a highly competitive salary... 

    Google Inc.

    Sunnyvale, CA
    4 days ago
  •  ...intersection of research, software engineering, hardware engineering, and...  ...end developer experience for ML development that, by taking...  ...acceleration, building ML compilers and runtimes for efficient execution...  ...packages. Experience with MLIR/LLVM or similar compiler... 

    Apple

    Cupertino, CA
    3 days ago
  •  ...leading tech firm in Sunnyvale is seeking a Senior Staff Architect to drive cutting-edge TPU technology for AI/ML applications. This role will focus on developing...  ...and a strong educational background in engineering. This position offers a competitive salary and comprehensive... 
    Worldwide

    Google

    Sunnyvale, CA
    3 days ago
  • Google is seeking a Staff Co-Design Engineer to shape the future of AI/ML hardware acceleration focused on TPU architecture and integration with large-scale AI/ML systems. You will work across model architecture and hardware design, driving innovative, high-performance... 

    Google

    Sunnyvale, CA
    3 days ago
  • Google is seeking a Staff SWE, Compiler Architect in Sunnyvale, California, to lead the development of performance modeling and optimization...  ...software/hardware integrations, you'll collaborate with top engineers and researchers to shape the future of advanced computing. Your... 

    Google

    Sunnyvale, CA
    3 days ago
  • About the Role We are seeking Inference Engineers to accelerate the performance of Pika's AI-driven products. In this highly technical...  ...including quantization, attention acceleration, and deep learning compiler stacks. GPU & Parallelism : Deep knowledge of GPU programming... 
    Work at office
    3 days per week

    Pika

    Palo Alto, CA
    3 days ago
  •  ...this role, you will report to a Senior Staff Engineering Manager Lead the design, development and...  ...Architect and implement scalable and robust ML pipelines for training, evaluating, and...  ...optimizing large scale models on GPU/TPU clusters for efficient serving Experience... 
    Local area
    Remote work

    Waymo

    Mountain View, CA
    1 day ago
  •  ...distillation, operator fusion, and model compilation (e.g., TensorRT, torch.compile, XLA)...  ...versions Collaborate closely with research engineers to translate model innovations into...  ...of experience in inference optimization, ML systems, or a closely related field Deep... 

    Rhoda AI

    Mountain View, CA
    19 hours ago
  • $204k - $259k

     ...collaborative group of machine learning (ML) engineers, software engineers, and ML research engineers...  ...this role, you will report to a Senior Staff Engineering Manager. You will: Lead the...  ...and optimizing large scale models on GPU/TPU clusters for efficient serving.... 
    Full time
    Remote work

    Waymo

    Mountain View, CA
    3 days ago
  • $138k - $198k

    Google is seeking a Hardware Architecture Modeling Engineer in Sunnyvale, California. In this role, you will drive the future of AI/ML hardware acceleration, working specifically on Tensor Processing Unit (TPU) technology. You'll develop custom silicon solutions, analyze... 

    Google

    Sunnyvale, CA
    3 days ago
  • Apple’s Server ML Frameworks team in Cupertino seeks engineers to optimize high-performance GenAI applications. This role focuses on ML inference framework projects and GPU programming while collaborating with cross-functional teams to enhance performance. The ideal candidates... 

    Apple

    Cupertino, CA
    3 days ago
  • Summary Apple’s Server ML Frameworks team in GPU, Graphics and...  ...data center. We are looking for engineers with systems background who...  ...parallelism. Develop kernel and compiler level optimizations and perform...  ..., CuTile, Triton, OpenXLA or LLVM is a plus Good understanding of... 

    Apple

    Cupertino, CA
    3 days ago
  • Google is seeking a Staff Software Engineer for AI/ML Infrastructure in Sunnyvale, CA. This role will involve leading the design and delivery of compute platforms at massive scale. Successful candidates will bring strong software development skills, particularly with languages... 

    Google

    Sunnyvale, CA
    3 days ago
  • $174k - $258k

    A leading technology company in Sunnyvale is seeking a Staff Hardware Systems Design Engineer to develop custom silicon solutions for AI Infrastructure. The role involves building next-generation hardware and software that impacts AI technology. Candidates must possess... 

    Google

    Sunnyvale, CA
    3 days ago
  •  ...The Mission: As a Senior Machine Learning Engineer, you will be responsible for building machine learning models/systems and innovative...  ...Science or a related field Experience with building and evolving ML Training and Inferencing systems at significant scale with a purpose... 
    Local area

    Typeface

    Palo Alto, CA
    19 hours ago
  • Google in Sunnyvale is seeking an experienced professional to shape the future of AI/ML hardware acceleration. You will work on cutting-edge TPU technology, driving innovation and developing custom silicon solutions. Your role involves maintaining emulation infrastructure... 

    Google

    Sunnyvale, CA
    3 days ago
  • $108.42k - $161.5k

     ...responsible for delivering and maintaining the tools and services engineers here at GM use every day to do their best work and drive our...  ...engineers at GM — how, where code is checked out, modified, compiled, tested, merged, and eventually deployed. Our goal is to ensure... 
    Work experience placement
    Work at office
    Local area
    Remote work
    Relocation
    Relocation package
    Flexible hours
    3 days per week

    General Motors

    Sunnyvale, CA
    3 days ago
  • Gravity Engineering Services Pvt Ltd. seeks experienced researchers and software engineers in Mountain View, CA to join the DUE Machine Learning team. The role focuses on developing advanced machine learning techniques aimed at evaluation systems for autonomous services... 

    Gravity Engineering Services Pvt Ltd.

    Mountain View, CA
    3 days ago
  • $213k - $263k

     ...simulation across 15+ U.S. states. The ML Optimization team at Waymo provides a set...  ...Research and Simulation. We are looking for engineers with ML software & systems expertise to...  ...extending and modifying underlying ML compilers and runtimes (e.g., OpenXLA/PjRT, TensorRT... 
    Full time
    Remote work

    Waymo

    Mountain View, CA
    3 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Staff ML Compiler Engineer - TPU/LLVM & MLIR. Be the first to apply!