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

Senior AI/ML Engineer for Ads Training & TPU Optimization

Google

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 to JAX, improve scalability, and drive cross-stack optimizations with robust debugging infrastructure.

The role emphasizes building high-performance ML pipelines, collaborating across teams, and delivering measurable improvements in efficiency and cost savings for ads processing at scale.

#J-18808-Ljbffr
Vacancy posted 14 hours ago
Similar jobs that could be interesting for youBased on the Senior AI/ML Engineer for Ads Training & TPU Optimization in Mountain View, CA vacancy
  • Google is seeking a software engineer to drive TPU efficiency at scale within the Ads Training Runtime team. You will lead efforts in model stability, hardware enablement...  ...of large-scale Ads models to JAX, while optimizing across stacks for performance and cost efficiency... 
    Senior
    Training

    Socket.dev

    Mountain View, CA
    2 days 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...  ...advanced audio-to-audio dialogue, TTS, and related AI capabilities. #J-18808-Ljbffr
    Senior
    Training

    Google

    Mountain View, CA
    14 hours ago
  • $207k - $300k

     ...infrastructure that powers our AI/ML applications. You will...  ...at the intersection of Ads, YouTube, and GenAI...  ...generation and optimization.Define the technical roadmap...  ...including creative data engines, generation and...  ...infrastructure to accelerate model training, evaluation, and rapid... 
    Training

    Google

    Mountain View, CA
    2 days ago
  •  ...Google in Mountain View is seeking a Senior Software Engineer for Creative Composition Intelligence to drive design, development, and productionization of advanced ML models that optimize Ad creative composition and delivery. You will lead architecture, mentor junior... 
    Senior

    Socket.dev

    Mountain View, CA
    14 hours ago
  •  ...Google is seeking a Senior Software Engineer in Creative Composition Intelligence to drive the design, development, and productionization of advanced ML models for optimizing Ad creative composition and delivery. You will lead efforts to accelerate innovation in automatically... 
    Senior

    Google

    Mountain View, CA
    14 hours ago
  • $182.4k - $250.6k

     ...driving? Join the Embodied AI team at General Motors....  ...real-world scenarios.As a Senior AI/ML Future Sensing Engineer in the Embodied AI organization...  ...such as unsupervised pre-training, imitation learning,...  ...debugging, and performance optimization for perception and sensor-... 
    Senior
    Training
    Full time
    Local area
    Remote work
    Work from home
    Relocation
    Relocation package
    Flexible hours

    General Motors

    Sunnyvale, CA
    2 days ago
  •  ...patients worldwide.We’re a team of engineers, clinicians, and innovators...  ...robotic platforms. As a Senior AI/ML Research Engineer (Computer...  ...vision layer: designing, training, and evaluating models that...  ...can build on, knowing when to optimize versus when to move fast, and... 
    Senior
    Training
    Local area
    Worldwide
    Flexible hours

    Intuitive Surgical

    Sunnyvale, CA
    10 hours ago
  • $170.6k - $261.3k

     ...About the team: The AV ML Infra team at GM...  ...the unique demands of AI and ML innovation, supporting...  ...productivity of ML engineers, and drive the...  ...Compute: Streamlines and optimizes large-scale ML training and inference across...  ...Position Overview: As a Senior AI/ML Full-Stack... 
    Senior
    Training
    Full time
    Local area
    Work from home
    Flexible hours

    General Motors

    Sunnyvale, CA
    1 day ago
  •  ...Google is seeking a Senior Software Engineer for AI/ML in Ads and Commerce, based in Mountain View, CA. You will design and implement ML features, leverage cutting-edge tech like Gemini and RAG, and contribute to scalable data pipelines and APIs. The role emphasizes... 
    Senior

    Google Inc.

    Mountain View, CA
    14 hours ago
  •  ...Google is seeking a Senior Software Engineer in Mountain View, CA, focused on AI/ML for Ads and Commerce. You will design and implement advanced ML features, leverage state-of-the-art infrastructure, and partner with cross-functional teams to deliver impactful products... 
    Senior

    Google

    Mountain View, CA
    14 hours ago
  • $260k - $330k

     ...recommendation, prediction, optimization, ads, marketplace,...  ...and distributed ML workflows. We need strong engineering skills in Python,...  ...examples, training data, and online serving...  ...Technologies: AI Support Java...  ...are hiring for a Senior Principal Machine... 
    Senior
    Training
    Full time
    Work at office
    Remote work

    PubMatic

    Redwood City, CA
    5 days ago
  • $195.2k - $262.2k

     ...infrastructure for the global AI economy. We are...  ...data and model training through to...  ...large in-house AI/ML infrastructure. Built by engineers, for engineers. From...  ...orchestration to inference optimization, we own the hard...  ...engineering. A Senior MLE owns substantial... 
    Senior
    Training
    Full time
    Temporary work
    Immediate start
    Remote work

    Nebius

    Palo Alto, CA
    1 day ago
  • $155.42k - $395.9k

     ...DescriptionAbout the Team:The ML Compute Platform is part of the AI Compute Platform...  ...development by optimizing for high-priority,...  ...supports the training and deployment of...  ...are looking for a Senior Software Engineer to join our team...  ...Experience with GPU/TPU optimizations... 
    Senior
    Training
    Full time
    Local area
    Remote work
    Work from home
    Relocation
    Relocation package
    Flexible hours

    General Motors

    Sunnyvale, CA
    2 days ago
  • $260k - $330k

     ...scale prediction or optimization systemsPubMatic is the leading AI-powered ad tech company...  ...are looking for a Senior Principal Machine Learning Engineer to help build the next...  ...candidate has strong ML fundamentals and experience...  ...negative examples, training datasets, and... 
    Senior
    Training
    Work at office
    Remote work

    PubMatic

    Redwood City, CA
    3 days ago
  • $204k - $259k

     ...environments for testing, training, and validation of...  ...machine learning (ML) engineers, software engineers...  ...will report to a Senior Staff Engineering...  ...and generative AI into robust, production...  ...in training and optimizing large scale models on GPU/TPU clusters for efficient... 
    Senior
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    2 days ago
  • $120k - $220k

     ...information powered by advanced AI, recommendation systems...  ...cycle.We're hiring the engineer who owns this agent end...  ...-ready 9:16 vertical ad — script, scene plan, b...  ...checkpoint — Train and ship on past CPI-winning...  ...them, trained a LoRA, optimized inference, built a... 
    Training
    Full time
    Local area
    Work from home

    News Break

    Mountain View, CA
    2 days ago
  •  ...Google in Sunnyvale seeks software engineers to develop next-generation AI and large-scale systems. You will contribute across the full stack, from model...  ...will collaborate on design reviews, implement scalable ML solutions, and help build products that impact billions... 
    Senior

    Google

    Sunnyvale, CA
    14 hours ago
  • $227k - $300k

     ...transformation to AI-enabled software-...  ...for a great Senior Staff AI Engineer to join our seasoned...  ...own the end-to-end ML pipeline—from...  ...ingestion and model training to deployment on...  ...devices and model optimization. You will work in...  ...Basic, Voluntary & AD&D)Unlimited paid... 
    Senior
    Training
    Work at office
    Worldwide
    Flexible hours
    Shift work
    3 days per week

    Sonatus

    Sunnyvale, CA
    2 days ago
  • $147.8k - $258.47k

     ...learn more. Within the Monetization Engineering team, we try to connect the dots between...  ...you will build systems that improve the ads delivery funnel, design experiments that...  ...Build and improve the ads marketplace to optimize value for Pinners, Partners and Pinterest... 
    Senior
    Relocation package
    Flexible hours

    Pinterest

    Palo Alto, CA
    14 hours ago
  • $174k - $253k

     ...Senior Software Engineer, AI/ML, Ads Training Mid Experience driving progress, solving problems, and mentoring...  .... In this role, you will drive TPU efficiency at scale for our core...  ...generation TPUs. By leading cross‑stack optimizations, robust debugging infrastructure,... 
    Senior
    Training
    Flexible hours

    Google

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

     ...environments for testing, training, and validation of...  ...machine learning (ML) engineers, software engineers...  ...will report to a Senior Staff Engineering...  ...and generative AI into robust, production...  ...in training and optimizing large scale models on GPU/TPU clusters for efficient... 
    Senior
    Training
    Full time
    Remote work

    Waymo

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

     ...more specialized ML areas, leverage ML...  ...PM, UX, and other engineering teams.Minimum qualifications...  ...model evaluation, optimization, data processing,...  ...).Generative AI experience....  ...: we believe that ads and commercial information...  ...education or training. US: $174000 - $25... 
    Senior
    Training
    Worldwide

    Google

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

     ...one or more specialized ML areas, leverage ML...  ...deployment, model evaluation, optimization, data processing,...  ...technologies.Google's software engineers develop the next-...  ...technology forward.Google Ads is helping power the...  ...relevant education or training. US: $174000 - $253000... 
    Senior
    Training

    Google

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

     ...environments, cloud infrastructure, and ML/AI GPU platforms for AV research and...  ...in GM. The Role: GM is looking for a Senior Capacity Engineer to join the AV Capacity and Performance...  ...cloud budget, identify and execute optimization projects, and provide capacity planning... 
    Senior
    Full time
    Work experience placement
    Work at office
    Local area
    Remote work
    Work from home
    Flexible hours
    3 days per week

    General Motors

    Sunnyvale, CA
    1 day ago
  • $150k - $225k

     ...quality of life. As a Senior AI / Embedded Engineer, you will be responsible for...  ..., model development, optimization, and deployment on embedded...  ...reliable, low-power, real-time ML systems that operate at...  ...collect, version, and manage training datasets at scale • Model... 
    Senior
    Training
    Full time
    Work at office
    Immediate start
    Visa sponsorship
    Night shift

    E-Space

    Saratoga, CA
    10 hours ago
  • $174k - $252k

     ...designs as we build the next-gen AI/LLM-powered Ads serving infrastructure.Identify performance optimization opportunities and deliver...  ...Automation.Google's software engineers develop the next-generation technologies...  ..., and relevant education or training. US: $174000 - $252000 (USD)... 
    Senior
    Training

    Google

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

     ...Economics, Operations Research, Engineering, or a related quantitative...  ...a PhD degree.Google Display Ads (GDA) is a network of...  ...systems that power all these optimizations.Specifically for our team, we...  ..., and relevant education or training. US: $174000 - $252000 (USD)... 
    Senior
    Training
    Work experience placement

    Google

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

     ...to ensure we design and optimize hardware and software...  ...will report to a Software Engineering Manager. You will:...  ...Leverage agentic AI to accelerate analysis,...  ...experience working with ML inference or linear algebra...  ..., experience, relevant training and education, and... 
    Senior
    Training
    Full time
    Remote work

    Waymo

    Mountain View, CA
    2 days ago
  • $165k - $205k

     ...information powered by advanced AI, recommendation systems...  ...Scientist to join our Ad Platform team. In this...  ...modeling to optimize ad delivery, improve targeting...  ...teams, including engineering, product, and sales, to...  ...relevant education or training. At NewsBreak, we design... 
    Senior
    Training
    Full time
    Local area
    Work from home

    News Break

    Mountain View, CA
    4 days ago
  •  ...SRI Tech SolutionsJob Title: AI / ML engineerLocation: Sunnyvale,...  ...skilled and experienced Senior Machine Learning Engineer / AI Solutions Architect with...  ...feature engineering, model training, evaluation, and deployment.Develop and optimize ML/DL models for NLP, computer... 
    Training
    Full time

    SRI Tech

    Sunnyvale, CA
    2 days ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Senior AI/ML Engineer for Ads Training & TPU Optimization. Be the first to apply!