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 Google

Vacancy posted 1 day 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
    1 day 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,...  ...audio-to-audio dialogue, TTS, and related AI capabilities. #J-18808-Ljbffr Google
    Senior
    Training

    Google

    Mountain View, CA
    2 days 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
    1 day 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
    1 day 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
    4 days 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
    14 hours ago
  • $144.7k - $261.3k

     ...cloud infrastructure, and ML/AI GPU platforms for AV research...  ...: GM is looking for a Senior Performance Engineer to join the AV Capacity...  ...budget, identify and execute optimization projects, and provide...  ...reliability of large-scale ML training and inference environments... 
    Senior
    Training
    Work at office
    Local area
    Remote work
    Work from home
    Flexible hours
    3 days per week

    General Motors

    Sunnyvale, CA
    5 days 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 leadership... 
    Senior

    Google Inc.

    Mountain View, CA
    1 day 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
    3 days ago
  • $169.78k - $338.69k

     ...developing Level 4 autonomous driving (AD) technology to make...  ...experienced and mission-driven Senior/Staff AI Infrastructure Engineer, Inference & Optimization to lead the performance tuning,...  ...Background in deep learning training frameworks (PyTorch) and practical... 
    Senior
    Training
    Full time

    DiDi Labs

    San Jose, CA
    2 days ago
  • $174k - $253k

     ...one or more specialized ML areas, leverage ML...  ...technologies.Google's software engineers develop the next-...  ...forward.The Ads Training Runtime team enables the...  ...role, you will drive TPU efficiency at scale for...  ...By leading cross-stack optimizations, robust debugging infrastructure... 
    Senior
    Training
    Flexible hours

    Google

    Mountain View, CA
    4 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
    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
    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

    Google Inc.

    Mountain View, CA
    3 days 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
    1 day ago
  • $174k - $253k

     ...Learning infrastructure, ML execution frameworks (...  ...JOB Google's software engineers develop the next-...  ...technology forward. The Ads Training Runtime team enables the...  ...role, you will drive TPU efficiency at scale...  ...By leading cross-stack optimizations, robust debugging infrastructure... 
    Senior
    Training
    Flexible hours

    Socket.dev

    Mountain View, CA
    1 day ago
  • $147.8k - $258.47k

     ...to 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
    1 day 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
    1 day ago
  • $241k - $326k

     ...centered on trust and optimized for culture,...  ...Overview: This role is a Senior Technical Leader (...  ...a team of 15-25 engineers. Solutions will vary...  ...domains such as Search, Ads, Notifications,...  ...Senior and Staff/Lead AI Engineers,...  ...foundational LLMs.Design and train large language... 
    Senior
    Training
    For contractors
    Work at office
    Flexible hours

    Linkedin

    Sunnyvale, CA
    4 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
    3 days 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
    14 hours 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
    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
    14 hours 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
    3 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
    1 day ago
  • $174k - $252k

    Design and implement scalable AI solutions for Google's enterprise functions leveraging...  ...building blocks.Collaborate with ML engineers and leads on developing reusable enterprise...  ..., experience, and relevant education or training. US: $174000 - $252000 (USD) + 15% bonus... 
    Senior
    Training

    Google

    San Jose, CA
    3 days ago
  •  ...Mountain View is seeking an experienced professional to enhance ML infrastructure for training workloads. Responsibilities include designing distributed...  ...data pipelines and collaborating with research teams to optimize systems. Candidates should have a strong background in... 
    Senior
    Training

    Waymo

    Mountain View, CA
    1 day ago
  • $120.8k - $193.3k

    DescriptionJob Title: Sr. Engineer, AI/ML Apps Engineering Job Location:...  ...Description: SiMa.ai is seeking a Senior Applications Engineer to...  ...architecture, AI software optimization, and technical leadership.You...  ...notes, tutorials, training materials, and customer enablement... 
    Senior
    Training
    Full time
    Work at office

    SiMa Technologies

    San Jose, CA
    2 days ago
  • $250k - $350k

    About the RoleWe are seeking Senior/Staff level Inference Engineers to accelerate the performance of Pika's AI-driven products. In this...  ...at scale.You will design and optimize inference pipelines, implement...  ...models into production.Improve Training Efficiency: (Bonus)... 
    Training
    Work at office
    3 days per week

    Pika

    Palo Alto, CA
    2 days ago
  • $147k - $211k

     ...is ready for analysis.Build optimization solutions encompassing targeting...  ..., Operations Research, Engineering, or a related quantitative field...  ...and digital media. The Brand ads team is responsible for designing...  ..., and relevant education or training. US: $147000 - $211000 (USD)... 
    Training
    Work experience placement
    Local area
    Immediate start

    Google

    Mountain View, 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!