{"schemaVersion":"jobsearcher.job.v1","id":"afdbece26cf32fa1796bf32a","url":"https://jobsearcher.com/jobs/afdbece26cf32fa1796bf32a","canonicalUrl":"https://jobsearcher.com/jobs/afdbece26cf32fa1796bf32a","title":"Senior AI/ML Systems Engineer TPU & JAX Efficiency","description":"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, and migration of large-scale Ads models to JAX, while optimizing across stacks for performance and cost efficiency.\r\nYou will contribute to robust debugging infrastructure, support next-generation TPUs, and help deliver significant efficiency gains and business impact for Google Ads.\r\nJ-18808-Ljbffr","company":"Socket","rawCompany":"socket","city":"Mountain View","state":"CA","isRemote":false,"isActive":false,"createdAt":"2026-08-08T01:38:16.104Z","occupations":[{"code":"15-1252.00","title":"Software Developers","slug":"software-developers"},{"code":"15-1299.08","title":"Computer Systems Engineers/Architects","slug":"computer-systems-engineers-architects"},{"code":"15-1221.00","title":"Computer and Information Research Scientists","slug":"computer-and-information-research-scientists"}],"industries":[{"code":"513210","title":"Software Publishers","slug":"software-publishers"},{"code":"541511","title":"Custom Computer Programming Services","slug":"custom-computer-programming-services"},{"code":"541512","title":"Computer Systems Design Services","slug":"computer-systems-design-services"}],"jobPosting":{"@context":"https://schema.org","@type":"JobPosting","title":"Senior AI/ML Systems Engineer TPU & JAX Efficiency","description":"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, and migration of large-scale Ads models to JAX, while optimizing across stacks for performance and cost efficiency.\r\nYou will contribute to robust debugging infrastructure, support next-generation TPUs, and help deliver significant efficiency gains and business impact for Google Ads.\r\nJ-18808-Ljbffr","datePosted":"2026-08-08T01:38:16.104Z","dateModified":"2026-08-08T01:38:16.104Z","hiringOrganization":{"@type":"Organization","name":"Socket","sameAs":"https://jobsearcher.com"},"jobLocation":{"@type":"Place","address":{"@type":"PostalAddress","addressLocality":"Mountain View","addressRegion":"CA","addressCountry":"US"}},"identifier":{"@type":"PropertyValue","name":"JobSearcher","value":"afdbece26cf32fa1796bf32a"},"url":"https://jobsearcher.com/jobs/afdbece26cf32fa1796bf32a"}}