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

Senior ML Performance Engineer - Fleet TPU

Google

Google Sunnyvale, CA, USA is seeking a Senior Software Engineer for Fleet-level ML Performance to drive end-to-end performance analysis of key ML workloads on future TPU systems, using advanced simulation tools and cross‑functional collaboration across the ML stack. The role emphasizes building scalable performance estimation methods in C++ and Python, balancing power, performance, and cost while addressing reliability and scheduling in data centers; you’ll influence TPU platform architecture #J-18808-Ljbffr Google

Vacancy posted 3 days ago
Similar jobs that could be interesting for youBased on the Senior ML Performance Engineer - Fleet TPU in Sunnyvale, CA vacancy
  • Google is seeking a Senior Software Engineer, AI/ML, Ads Training in Mountain View to advance core training systems and TPU efficiency. You will lead initiatives to migrate Ads models...  .... The role emphasizes building high-performance ML pipelines, collaborating across... 
    Senior
    Performance

    Google

    Mountain View, CA
    1 day ago
  • Google is seeking a software engineer to drive TPU efficiency at scale within the Ads Training Runtime team. You will lead efforts in model...  ...scale Ads models to JAX, while optimizing across stacks for performance and cost efficiency. You will contribute to robust debugging... 
    Senior
    Performance

    Socket.dev

    Mountain View, CA
    1 day ago
  • $174k - $253k

    Senior Software Engineer, ML Compilers, TPU, YouTube corporate_fare YouTube place San Bruno, CA, USA ; Mountain...  ...Experience with debugging correctness and performance issues at all levels of the ML...  ...Pallas kernels in JAX, maximizing fleet utilization and the value... 
    Senior
    Performance
    Fleet

    Google Inc.

    Mountain View, CA
    3 days ago
  • $174.72k - $295.68k

     ...connectivity. We are seeking Machine Learning Engineers with strong expertise in generative...  ...multimodal data from our autonomous fleet and robotics platforms. Job Responsibilities...  ...Vision-Language-Action (VLA) driving performance. Extend prediction beyond 2D pixel... 
    Senior
    Performance
    Fleet
    Full time

    XPENG Motors

    Santa Clara, CA
    1 day ago
  • Google is seeking a Senior Software Engineer to optimize ML models powering the YouTube algorithm on TPU hardware. You will contribute to the XLA compiler, author custom kernels in JAX, and maximize fleet utilization while delivering value to users. You will work on new... 
    Senior
    Fleet

    Google Inc.

    Mountain View, CA
    3 days ago
  • $193.93k - $291.15k

     ...robotaxis and logistics fleets to personal vehicles....  ...looking for a Software Engineer to join our Sensor Data...  ...requirementsRole is scoped as a Senior/Staff IC with the...  ...understanding of ML fundamentals with hands...  ...eligible for an annual performance bonus, equity, and a competitive... 
    Senior
    Performance
    Fleet
    Immediate start
    Flexible hours

    Nuro

    Mountain View, CA
    4 days ago
  • $174k - $252k

    Perform fleet-level performance analysis of key ML workloads (e.g., Gemini) on future TPU systems using advanced simulation tools to evaluate hardware/software trade-offs and...  ...researchers, compiler developers, systems engineers, and TPU architects to analyze and optimize... 
    Senior
    Performance
    Fleet
    Worldwide

    Google

    Sunnyvale, CA
    1 day ago
  • $188.5k - $282.7k

     ...Rubrik's Semantic AI Governance Engine, which is the first system...  ...lead even further.As an Applied ML Engineer on the SAGE team, you...  ...traffic patterns.Engineering High-Performance Model Serving and Inference...  ...Global 2000 customer agent fleets.Optimizing live deployments through... 
    Senior
    Performance
    Fleet
    Permanent employment
    Local area

    Rubrik

    Palo Alto, CA
    23 hours ago
  • $189k - $300k

     ...works on and delivers ML models to the product that...  ...impacting AV product performance through smart use of data...  ...by GM and retail fleets, but also synthetic sim...  ...high-impact team of AI/ML engineers, data scientists and engineers...  ..., you will serve as a senior individual contributor... 
    Performance
    Fleet
    Full time
    Local area
    Remote work
    Work from home
    Relocation
    Relocation package
    Flexible hours

    General Motors

    Sunnyvale, CA
    23 hours ago
  • $153.2k - $234.1k

     ...autonomous vehicle behavior across real-world scenarios. As a Senior ML Infra Engineer, you will work on the core systems that enable rapid...  ...next. You will develop model training pipelines that are performant, easy to use, and exceptionally reliable. Your success will... 
    Senior
    Performance
    Full time
    Local area
    Remote work
    Work from home
    Relocation package
    Flexible hours

    General Motors

    Sunnyvale, CA
    4 days ago
  • $155.42k - $205.9k

     ...machine learning models, with a focus on performance, availability, concurrency, and scalability...  ...development by prioritizing high-impact, ML-centric use cases. About the Role: We are seeking a Senior ML Infrastructure engineer to help build and scale robust Compute platforms... 
    Senior
    Performance
    Full time
    Local area
    Remote work
    Work from home
    Relocation
    Relocation package
    Flexible hours

    General Motors

    Sunnyvale, CA
    23 hours ago
  •  ...Job Description Job Description: We are looking for a senior ML infrastructure engineer to build and evolve the systems that support model training...  ...Solid understanding of heterogeneous computing and high-performance computing, especially for ML training or serving... 
    Senior
    Performance

    Maxinsights Corporation

    Santa Clara, CA
    22 days ago
  • $153.2k - $234.1k

     ...infrastructure that powers every machine learning engineer working on our cutting-edge Autonomous...  ...entirely on enhancing the safety and performance of the car, rather than managing...  ...most advanced driverless vehicles.As a Senior ML Infra Engineer, you will build critical... 
    Senior
    Performance
    Full time
    Work at office
    Local area
    Remote work
    Work from home
    Relocation
    Relocation package
    Flexible hours

    General Motors

    Sunnyvale, CA
    23 hours ago
  •  ...AI lab in Santa Clara is seeking a skilled software engineer with over 8 years of experience to optimize machine...  ...distributed training strategies, collaborating with ML researchers, and developing tools for performance enhancement. Ideal candidates have a deep understanding... 
    Senior
    Performance

    Odyssey

    Santa Clara, CA
    4 days ago
  • $174.9k - $261.3k

     ...understand the world!The Data Labeling Engineering team designs, builds, and operates hybrid...  ...software engineering, data engineering, and AI/ML, defining the strategies, tooling, and...  ..., implement, and test scalable, high‑performance user experiences and services using... 
    Senior
    Performance
    Full time
    Local area
    Remote work
    Work from home
    Relocation package
    Flexible hours

    General Motors

    Sunnyvale, CA
    1 day ago
  • $163k - $236k

     ...responsible for Silicon validation of TPU (Tensor Processing Unit - Google’s custom ML accelerator) design in mobile...  ...framework.Develop test plan, perform pre-silicon validation (Emulation...  ...:Bachelor's degree in Electrical Engineering, Computer Engineering, Computer Science... 
    Senior
    Performance
    Worldwide

    Google

    Mountain View, CA
    23 hours ago
  • Google DeepMind is hiring a Senior Software Engineer for Gemini Audio in Mountain View. The role focuses on building and optimizing training infrastructure for audio models running on TPUs, with emphasis on Accelerated Linear Algebra (XLA) and efficient model training... 
    Senior

    Google

    Mountain View, CA
    2 days ago
  • $144.7k - $261.3k

    Job DescriptionThe Senior ML Validation Research Engineer will lead applied machine learning research focused on improving verification and validation...  ....Key ResponsibilitiesPrototype research concepts into performant tools integrated into CI/CD and large-scale validation... 
    Senior
    Performance
    Full time
    Local area
    Remote work
    Work from home
    Flexible hours

    General Motors

    Sunnyvale, CA
    4 days ago
  • Google Cloud's TPU Chip Architecture and Performance team analyzes ML workloads to balance performance and cost on next-generation TPU systems. You will work...  ...model researchers, compiler developers, and systems engineers to drive architecture decisions and performance... 
    Senior
    Performance

    Socket

    Sunnyvale, CA
    3 days ago
  •  ...firm in Sunnyvale is seeking a Senior Staff Architect to drive cutting-edge TPU technology for AI/ML applications. This role will...  ...domain experts to enhance performance and efficiency. Candidates must...  ...strong educational background in engineering. This position offers a... 
    Senior
    Performance
    Worldwide

    Google

    Sunnyvale, CA
    1 day ago
  • Waymo is seeking a seasoned ML/Computer Vision Engineer to advance the Waymo Driver stack. You will own ML tasks, optimize performance, and scale solutions for real-world autonomous driving. The role requires deep learning expertise and hands-on Python with PyTorch or... 
    Senior
    Performance

    Neura Market

    Mountain View, CA
    1 day ago
  • Google Cloud is seeking a Senior Performance Co-Design Engineer for TPU work in Sunnyvale to optimize serving performance of large language models on custom AI silicon. You will analyze current and emerging models, build tools to probe bottlenecks, and influence architecture... 
    Senior
    Performance

    Google

    Sunnyvale, CA
    2 days ago
  • $193.3k - $261.5k

     ...MLIL DataPlane team is looking for a Senior Software Development Engineer to own the design and implementation...  ...custom hardware, developing high-performance compute kernels, enabling efficient...  ...optimize compute kernels for a custom ML accelerator architecture, targeting... 
    Senior
    Performance
    Local area
    Flexible hours

    Amazon

    Cupertino, CA
    23 hours ago
  • Rhoda AI in Mountain View is seeking a Staff / Principal ML Training Systems Engineer to lead the performance of large-scale multimodal training systems. This role involves improving training efficiency and collaborating closely with research teams to accelerate model... 
    Senior
    Performance

    Rhoda AI

    Mountain View, CA
    4 days ago
  • $235.03k - $352.29k

     ...applications, from robotaxis and commercial fleets to personally owned vehicles. With...  ...driving technology. In an ML-first system, the overall system performance depends heavily on the quantity...  ...closely collaborates with autonomy engineers to ensure our labeled data is high... 
    Senior
    Performance
    Fleet
    Full time

    Nuro

    Mountain View, CA
    23 hours ago
  • CoreWeave is seeking a Senior Storage Engineer, File & Block, to design, build, and operate scalable file and block storage for a high-performance, multi-tenant environment. You will own the data...  ...storage, running on our hardware fleet and delivered through Kubernetes. Ideal... 
    Senior
    Performance
    Fleet

    Socket.dev

    Sunnyvale, CA
    1 day ago
  • $122.6k - $185k

     ...delivering continuous price performance improvements in the cloud for...  ...performance and scalability in AI/ML and HPC workloads.You are...  ...builders like you. The AWS Hardware Engineering team creates server designs...  ...launching hardware in the fleet. Located out of Seattle,... 
    Performance
    Fleet
    Local area
    Flexible hours

    Amazon

    Cupertino, CA
    3 days ago
  • $160.36k - $240.54k

     ...applications, from robotaxis and commercial fleets to personally owned vehicles. With...  ...foundational infrastructure where correctness and performance have consequences on a real road.About...  ...operationsAbout You4+ years of backend engineering experience with strong Go... 
    Senior
    Performance
    Fleet

    Nuro

    Mountain View, CA
    2 days ago
  • $182.4k - $250.6k

     ...autonomous vehicle behavior across real-world scenarios.As a Senior AI/ML Future Sensing Engineer in the Embodied AI organization, you will develop and...  ...sensing architecture decisions and autonomous driving performance. You will contribute to designing and improving ML and... 
    Senior
    Performance
    Full time
    Local area
    Remote work
    Work from home
    Relocation
    Relocation package
    Flexible hours

    General Motors

    Sunnyvale, CA
    1 day ago
  • $193.93k - $291.15k

     ...range of applications, from robotaxis and commercial fleets to personally owned vehicles. With technology...  ...FEC (Forward Error Correction) algorithms that out-perform standard protocols. About the Work Engineered Connectivity: Architect a network bonding... 
    Senior
    Performance
    Fleet
    Full time
    Remote work

    Nuro

    Mountain View, CA
    23 hours ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Senior ML Performance Engineer - Fleet TPU. Be the first to apply!