Senior ML Compiler Engineer for TPUs
Google is seeking a Senior Software Engineer to advance ML compiler technology for TPUs and YouTube-scale recommendations. You will contribute to the Accelerated Linear Algebra (XLA) compiler, author custom kernels in JAX, and optimize models for TPU hardware across the YouTube ecosystem. You will engage with hardware designers to co-design future processors and push performance at distributed scale, while shaping end-to-end ML software from model design to deployment. #J-18808-Ljbffr Google
Vacancy posted 4 days ago
Similar jobs that could be interesting for youBased on the Senior ML Compiler Engineer for TPUs in San Bruno, CA vacancy
$128.7k - $261.3k
...mobility. For the AI Kernels & Compilers team, that mission shows up... ...development, and performance engineering so that every cycle on our accelerators... ...driving. The RoleAs a Senior Compiler Engineer on the AI... ...reliable, and effortless for ML engineers across the AV...SeniorFull timeLocal areaRemote workWork from homeRelocation packageFlexible hours- ...robotics and autonomous vehicles, femtoAI brings the power of AI to everyday devices. Job Description You will work on a custom ML compiler that transforms modern ML and DSP models into highly efficient programs for our accelerator. This role spans the full compiler stack...Suggested
- ...specializing in AI solutions, based in San Bruno, is looking for a Compiler Developer. The role involves building and maintaining model... ...implementing graph transformations, and optimizing execution for a custom ML compiler. Candidates should have over 2 years of experience in...Suggested
- A technology company based in San Bruno is seeking a Compiler Engineer to work on a custom ML compiler for their AI accelerator. The role requires a minimum of 2 years experience in compilers or edge-AI, with strong proficiency in Python and/or C++. You will be responsible...Suggested
$170.1k - $258.3k
...accessible mobility. For the AI Kernels & Compilers team, that mission shows up in the... ...export, kernel development, and performance engineering so that every cycle on our accelerators... ...that sit at the heart of our on‑vehicle ML inference for ADAS and autonomous driving...SeniorFull timeLocal areaRemote workWork from homeRelocation packageFlexible hours$166k - $210.25k
RDQ127R59SummaryAs a Senior Applied ML Engineer on the Applied AI team at Databricks, you will use machine learning, scheduling, and optimization... ...the entire stack—from cluster management down to query compilation. You will solve complex, high-impact engineering problems...SeniorLocal areaWorldwide- Position: Senior ML Performance Engineer Location: SF Bay Area (US) or Toronto (Canada) - Hybrid Employment Type: Full-Time Industry: AI Infrastructure / Compiler Systems Overview A venture-backed AI infrastructure company is building a high-performance, portable compiler...SeniorFull time
- ...governance, maintain auditability, and deliver reliable outcomes at scale. About the Role We are looking for a visionary Senior ML Engineer who will bridge the gap between high-level architecture and hands-on execution, specifically focusing on simplifying...SeniorFull timeShift work
Palm Venture Studios
San Francisco, CA1 day ago - ...are looking for an experienced software engineer to help us build a new generation of transpilation... ...backgrounds for this role: Path A — Compiler / Tooling You have built a compiler,... ...frameworks. Familiarity with ML tools and frameworks (PyTorch, Hugging Face...SeniorFull timeRemote workRelocation packageFlexible hours
Code Metal
San Francisco, CA1 day ago $168.1k - $312.3k
...new insights into disease mechanisms, therapeutic opportunities, and human biology. We seek a highly motivated and passionate Senior ML Engineer to join our Generative Modeling team and help build and scale foundation models and agentic systems for therapeutic discovery...SeniorFull timeLocal areaWorldwideRelocation package$147.6k - $274k
...techniques. We are seeking a highly motivated and skilled ML Infrastructure DevOps Engineer to join our growing team within Genentech Research and... ...and gaps, proposing ideas and suggestions for improvements.Senior Machine Learning Engineer - DevOpsLead the architecture and...SeniorFull timeLocal areaWorldwideRelocation package$200k - $280k
...relics of the past. Our team is a tight-knit group of bold thinkers—engineers, innovators, and industry experts—who thrive on turning... ...is your shot. About the Role We're looking for a skilled ML engineer to build the perception systems that give our autonomous...SeniorFull timeTemporary workFlexible hoursShift workAgtonomy
South San Francisco, CA1 day ago$170k - $240k
...independent, self-motivated and be expected to explore relevant ML literature and apply novel approaches to various protein design... ...in TensorFlow, PyTorch, and/or JAXStrong software engineering skills and substantial expertise in PythonTrack record of outstanding...Senior- Google Inc. is seeking a Senior Software Engineer specializing in AI/ML for YouTube in San Bruno, California. This role involves developing technologies that change user interaction and requires at least 5 years of programming experience in Python and C++. Candidates should...Senior
$174k - $253k
Senior Software Engineer, ML Compilers, TPU, YouTube Note: By applying to this position you will have an opportunity to share your preferred working location... ...stack. Familiarity with accelerator HW architectures (TPUs/GPUs). About the job Google's software engineers...Senior$77k - $202k
...generation analytics platform tailored for healthcare payers. As a Senior Associate, you will analyze complex problems, mentor junior... ...Bachelor's Degree- At least 4 years of experience in software engineering or data engineeringWhat Sets You Apart- Master's Degree in...SeniorFull timeH1b- ...who want to push the frontier of what’s possible. As a Senior Machine Learning Engineer , you will play a leading role in designing the next... ...and an interest in applying cutting-edge breakthroughs in ML to meaningful problems in drug discovery. We are looking for...SeniorWork at officeVisa sponsorshipFree visa3 days per week
Tahoe Therapeutics
South San Francisco, CAmore than 2 months ago $190k - $205k
...acoustic, magnetic, vibration, electrical, and visual signals Production Engineering Write clean, scalable, well-tested Python code that integrates into a large shared codebase. Build end-to-end ML pipelines including data processing, feature extraction, training,...SeniorFull timeLive inGridware
San Francisco, CA1 day ago$162.8k - $203.5k
...platform for low-cost, ultra-immersive transportation to improve people’s lives warrants modern ML utilizing peta-byte scale data. Our highly motivated Machine Learning Engineers work on these challenging problems and define solutions to directly impact various aspects of...SeniorHourly payFull timeWork experience placementWork at officeLocal area3 days per weekLyft
San Francisco, CA1 day ago$162.8k - $203.5k
...and compete with companies that have been in the mapping business for decades. To strengthen our efforts, we are hiring a Senior ML Engineer who will work end-to-end on creating and improving new capabilities to detect changes in the environment and reflect them in...SeniorHourly payFull timeWork experience placementWork at officeLocal area3 days per weekLyft
San Francisco, CA1 day ago- An innovative company is seeking a Distributed Systems/ML Engineer to enhance the training throughput of its internal framework. This role involves collaborating with researchers to develop efficient video models and applying cutting-edge techniques to optimize training...Senior
$161.93k - $227.33k
...Senior Machine Learning Engineer Brisbane, California About this opportunity: At Freenome, we are seeking a Senior Machine Learning Research... ...reliable, scalable artificial intelligence/machine learning (AI/ML) systems in a cloud environment. The MLS team at...SeniorWork at officeLocal areaRemote work2 days per week3 days per weekFreenome
Brisbane, CA1 day ago$200k - $280k
A leading voice AI startup is seeking a Founding Senior Machine Learning Engineer to fine-tune and deploy human-like voice agents, handling millions... ...Candidates should have real-world experience in deploying ML models and work well in a fast-paced environment. #J-18808...Senior- Perplexity AI Inc. is seeking experienced ML Engineers to design and optimize core recommendation systems, enhancing user experience. Successful candidates will have hands-on experience in production environments developing personalization, ranking, and user modeling systems...Senior
- Reddit is hiring Machine Learning Engineers (IC4) to build and evolve the auction, bidding and budgeting systems powering... ...Optimization, Product, Data Science and Infra. This senior role requires 3-5+ years in production ML, strong Python/Java/Go skills, and experience with...Senior
- ...GPUs, optimizing throughput and stability while advancing production-ready training pipelines. This role emphasizes collaboration with ML researchers, debugging across GPU stacks, and improving communication and memory efficiency in large-scale training environments. #J...Senior
$160k - $240k
Tensec is looking for a Machine Learning Engineer in San Francisco to build algorithms and optimization systems that drive their autonomous decision engine. This role involves designing trading strategies, building execution layers, and deploying robust models that enhance...SeniorRelocation package$190k - $265k
Harvey, Inc in San Francisco is looking for a Research Engineer to enhance their AI platform for law, tax, and finance. This role will involve... ...documents. The ideal candidate will have over 5 years of ML engineering experience and proven ability in advanced research....Senior- A cutting-edge AI technology company based in San Francisco is seeking a specialist to design and operate large-scale GPU infrastructure. This role requires expertise in deploying GPU systems for high-throughput inference and model performance optimization. The ideal candidate...Senior
- Aurelius Systems in San Francisco is seeking a Senior ML Engineer to lead the perception stack for Archimedes—detection, tracking, classification, and sensor fusion across the targeting chain. You will own perception as a domain and partner with the CTO and robotics software...Senior
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to Senior ML Compiler Engineer for TPUs. Be the first to apply!
Related searches
- senior manager clinical operations San Bruno, CA
- senior leadership San Bruno, CA
- senior grant accountant San Bruno, CA
- senior manager automotive San Bruno, CA
- senior environment artist San Bruno, CA
- senior magento developer San Bruno, CA
- senior application administrator San Bruno, CA
- senior manager accenture San Bruno, CA
- senior ux writer San Bruno, CA
- senior director digital marketing San Bruno, CA