Staff ML Compiler Engineer - TPU/LLVM & MLIR
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
$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)....SuggestedLocal areaFlexible hours- ...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
$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- 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...SuggestedFull time
- 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
- 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...
$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 timeTemporary workWorldwide$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,...- 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...
$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...- 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...
- 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...
- ...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...
- ...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 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 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...
- 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 office3 days per week
- ...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 areaRemote work
- ...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...
$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 timeRemote work$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...- 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...
- 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...
- 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...
$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...- ...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
- 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...
$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 placementWork at officeLocal areaRemote workRelocationRelocation packageFlexible hours3 days per week- 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...
$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 timeRemote work
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!
- technology administrator Mountain View, CA
- engineering aide Mountain View, CA
- senior staff systems engineer Mountain View, CA
- staff data engineer Mountain View, CA
- software engineer staff Mountain View, CA
- senior staff engineer Mountain View, CA
- assistant engineer Mountain View, CA
- staff engineer Mountain View, CA
- data scientist machine learning engineer Mountain View, CA
- machine learning engineer Mountain View, CA

