EdgeTPU ML Compiler Engineer — Build High-Performance ML
Google is seeking a Senior Software Engineer to join the EdgeTPU Compiler team, focusing on next-generation compiler optimizations for ML models on custom hardware. You will translate framework-level code into efficient EdgeTPU instructions and own end-to-end features. You will triage performance and correctness issues, collaborate with ML model owners to optimize architectures for EdgeTPU, and drive production-grade compiler improvements across projects. #J-18808-Ljbffr Google
Vacancy posted 3 days ago
Similar jobs that could be interesting for youBased on the EdgeTPU ML Compiler Engineer — Build High-Performance ML in Mountain View, CA vacancy
- Waymo is seeking a seasoned software engineer to advance our compute stack for the Waymo Driver. Hybrid schedule... ...a Software Engineering Manager, you will shape compiler support, performance tuning, and NN inference on our high-performance hardware. You will collaborate with...Performance
$207k - $300k
Deliver certified performance benchmarks six months... ...pilot goals under compiler constraints.... ...technical proficiency ML workload... ...familiarity with high level compiler intermediate... ...Google's software engineers develop the next-generation... ...developers to build the future. From...PerformanceWorldwideGoogle
Mountain View, CA1 day ago$213k - $263k
...Waymo has focused on building the Waymo Driver-The World... ...architect and create high-performance custom silicon; we... ...report to a Software Engineering Manager. You will: Leverage... ...our production grade compiler, and design and... ...experience working with ML inference or linear algebra...PerformanceFull timeRemote work$165.2k - $223.6k
The AWS Neuron Compiler team is actively seeking skilled compiler engineers to join our efforts in developing... ...for advanced ML capabilities, powering... ...to ensure system-wide performance optimization. As part... ...defects. You’ll also:Build high-impact solutions to deliver...PerformanceInternshipLocal areaFlexible hoursAmazon
Cupertino, CA6 days ago$128k - $256k
...for the computational performance optimization of ByteDance... ...and development of high-performance kernel libraries... ...cultivate model compilation optimization technology... ...and code generation, build and improve the mid-platform... ...Science, Computer Engineering, or a related...PerformanceTemporary workLocal areaByteDance (old)
San Jose, CA21 hours ago$184.7k - $324.8k
...On-Device ML Compiler Engineer, Model Compilation, Graphics, Games and Machine... ...innovation. Our team builds the essential infrastructure... ...model authoring, runtime, and performance teams to ensure that models... ...use cases. We’re seeking a highly motivated software engineer...PerformanceRelocationApple Inc.
Cupertino, CA1 day ago$250k - $350k
...Senior/Staff level Inference Engineers to accelerate the performance of Pika's AI-driven products. In this highly technical role, you will... ...acceleration, and deep learning compiler stacks.GPU & Parallelism:... ...office culture—we’re all building and launching togetherAbout...PerformanceWork at office3 days per weekPika
Palo Alto, CA2 days ago$165.2k - $223.6k
...the SDK that optimizes the performance of complex ML models executed on AWS... ...This role is for a software engineer in the Compiler team for AWS Neuron. As... ...will be responsible for building next generation Neuron compiler... ..., making Neuron compiler highly performant and easy-to-...PerformanceInternshipLocal areaFlexible hoursAmazon
Cupertino, CA6 days ago$180k
...the stars. SpaceXSI is building Super Intelligence to help... ...ABOUT THE ROLE: As an ML Infrastructure Engineer, you will play a pivotal role... ...optimizing the reliable, high-performance ML platform that powers... ...Python and experience with compiled languages such as C++ or Rust...PerformanceTemporary workWork experience placementSpaceXSI
Palo Alto, CA1 day ago$228k - $285k
...Electric Adventure Vehicles we build, and the curious, courageous... ...will be a key member of the ML Compiler team working on software... ...implementation targeting for performance. You will help bring up new... ...Ph.D. or M.S. in Computer Engineering or a related field. Excellent...PerformanceFull timeContract workTemporary workPart timeLocal areaShift workWatershed
Palo Alto, CA1 day ago- Apple is seeking an ML Infrastructure Engineer to advance on-device ML by building graph compilers and runtimes for Apple devices. You will shape the infrastructure enabling... ..., and execution on Apple Silicon, pushing performance and efficiency in constrained environments....Performance
- Cerebras Systems builds the world's largest... ...workloads with ultra high-speed inference.... ...The RoleThe Core ML team develops novel... ...Wafer-Scale Engine. Our work spans efficient... ...ML frameworks, compilers, runtimes, and low... ...capabilities, diagnose performance bottlenecks, and...Performance
Cerebras Systems
Sunnyvale, CA5 days ago - ...'d help us solve. We build the embodied intelligence... ...into that. About Our Engineering Teams The Performance Architecture team is... ...Working On Profiling ML workloads across... ...utilisation, e.g. parallelism, compilation, mixed precision,... ...way You write high quality, well-structured...PerformanceWork at officeRemote workVisa sponsorshipRelocation packageFlexible hoursShift work
$152k - $241.5k
...Machine Learning Applications and Compiler Engineer!NVIDIA is seeking engineers to... ...innovative!What you’ll be doing:Build, develop, and maintain high-performance runtime and compiler components,... ...spatial accelerators at top tier ML, compiler, and computer architecture...PerformanceFull timeNvidia
Santa Clara, CA6 days ago$170.6k - $261.3k
...the tooling this team builds. About the Role... ...mathematically rigorous engineer to own model numerics:... ...differences from quantization, compilation, and precision... ...make the math run . High proficiency in PyTorch... ...payouts based on company performance, job level, and...PerformanceLocal areaWork from homeFlexible hoursGeneral Motors
Sunnyvale, CA2 days ago- ...from prototyping new initiatives to building highly-scalable enterprise-grade AI products... ...s of developers and enterprise users.ML performance, quality, and systems acumen-ship: Experience... ...7+ years of software engineering experience building and operating enterprise...PerformanceWork at office
Atlassian
Mountain View, CA5 days ago $193.3k - $261.5k
...Amazon Web Services (AWS) builds AWS Neuron, the software... ...forefront of maximizing performance for AWS's custom ML accelerators. Working at... ...hardware-software boundary, our engineers craft high-performance kernels for... ...toolkit includes an ML compiler, runtime, and application...PerformanceInternshipLocal areaWork from homeFlexible hoursAmazon
Cupertino, CA2 days ago$139k - $223k
...backgrounds who are ready to help build a transportation ecosystem that... ...for all.We are seeking a ML Engineer to join the Behavior Planning Team. This is a high-impact role where you will sit... ..., with a focus on writing high-performance, maintainable code.Deep experience...PerformanceWork at officeLocal area3 days per weekAurora Innovation
Mountain View, CA4 days ago$173k - $253k
Matterport - Senior ML Ops Engineer Job Description CoStar Group is... ...computing platform turns buildings into data making every space... ...be pivotal in enhancing the performance, efficiency, and scalability... ...optimization techniques, and deploying highly efficient models into...PerformanceFull timeWork at officeWork from homeMatterport
Sunnyvale, CA5 days ago$178.42k - $230.5k
...the tools and services engineers here at GM use every... ...checked out, modified, compiled, tested, merged, and eventually... ...expected to identify high ROI investments with... ...to contribute code, perform code reviews, and... ...tools effectively. Building tools to enable engineers...PerformanceFull timeWork experience placementLocal areaWork from homeRelocation packageFlexible hoursGeneral Motors
Sunnyvale, CA3 days ago$150.4k - $277.6k
...Displays org provides a suite of high-performance data parallel algorithms for... ...and GPU programming engineers who are passionate about providing... ...in-depth analysis, compiler and kernel level optimizations... ...perspectives as possible, we help you build a career where you feel like...PerformanceRelocationApple Inc.
Cupertino, CA1 day ago$500 per month
...is an AI technology company building super-intelligence for the interactive... ...move that metric. As an ML Engineer on our AI team, you'll own... ...on it Are a builder with high agency and a bias for action,... ..., meaningful equity, and performance bonus. ~100% employer-paid...PerformanceWork at officeFlexible hoursAstrocade
Palo Alto, CA1 day ago$153.2k - $234.1k
...world scenarios. As a Senior ML Infra Engineer, you will work on the core... ...training pipelines that are performant, easy to use, and exceptionally... ...scalable, intuitive, and high‑performance training platforms... ...environments. Experience building reliable, high-performance,...PerformanceFull timeLocal areaRemote workWork from homeRelocation packageFlexible hoursGeneral Motors
Sunnyvale, CA4 days ago$207k - $300k
Build advanced AI-generated content (AIGC) quality frameworks and... ...through advanced context engineering and agentic feedback loops to... ...and feedback to improve the performance of the AIGC stack.Resolve key... ...low-latency surfaces.Navigate high ambiguity, drive technical innovation...PerformanceGoogle
Mountain View, CA6 days ago$124k - $250k
...intentional about the team and culture we are building, seeking candidates who are... ...in the LifeAs a member of our software engineering infra team, you'll solve technical challenges... ...infrastructure. The team builds a high-performance, high availability, globally...PerformanceAppLovin
Palo Alto, CA6 days ago$155.42k - $205.9k
...machine learning models, with a focus on performance, availability, concurrency, and... ...innovation and development by prioritizing high-impact, ML-centric use cases. About the Role:... ...seeking a Senior ML Infrastructure engineer to help build and scale robust Compute platforms...PerformanceFull timeLocal areaRemote workWork from homeRelocationRelocation packageFlexible hoursGeneral Motors
Sunnyvale, CA5 days ago- ...online and in-store. We are building and designing the next generation... ...in? We need solid prompt engineers with the talent and expertise... ...prompts, over time, based on performance metrics, user feedback and... ...especially those present in highly virtualized, elastic, cloud...Performance
BayOne Solutions
Sunnyvale, CA2 days ago $179k - $223k
...Electric Adventure Vehicles we build, and the curious, courageous souls... ...Autonomy stack. In this Senior ML Engineer role, you will play a key role in delivering high-quality, scalable auto-labeling... ...definition, evaluation, model performance optimization, feedback loop. A...PerformanceFull timeContract workLocal area$228k - $285k
...emissions-free Electric Adventure Vehicles we build, and the curious, courageous souls we... ...Role Summary As a Staff Software Engineer for ML Optimization and Hardware Acceleration... ...Transformers, LLMs, VLMs, LDMs and high-performance hardware execution. You will bridge...PerformanceFull timeContract workLocal areaRivian
Palo Alto, CA1 day ago$160.5k - $240.7k
..., Inc. Job Area Engineering Group, Engineering Group... ...optimize, and deploy ML models on Qualcomm... ...Workbench lets developers compile trained PyTorch or... ...on Snapdragon SoCs Build and maintain ONNX‑based... ...accuracy regressions, and performance degradations; develop...PerformanceQualcomm
Santa Clara, CA1 day ago
Do you want to receive more vacancies?
Subscribe and receive similar vacancies to EdgeTPU ML Compiler Engineer — Build High-Performance ML. Be the first to apply!
Related searches
- senior ml engineer Mountain View, CA
- machine learning ai engineer Mountain View, CA
- machine learning software engineer Mountain View, CA
- ai ml engineer Mountain View, CA
- machine learning engineer Mountain View, CA
- computer vision machine learning engineer Mountain View, CA
- senior performance engineer Mountain View, CA
- performance improvement specialist Mountain View, CA
- performance specialist Mountain View, CA
- high performance computing engineer Mountain View, CA