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

Senior ML Compiler Engineer for TPUs

Google

Google is seeking a Senior Software Engineer to advance ML compiler technology for TPUs and YouTube-scale recommendations. You will contribute to the Accelerated Linear Algebra (XLA) compiler, author custom kernels in JAX, and optimize models for TPU hardware across the YouTube ecosystem. You will engage with hardware designers to co-design future processors and push performance at distributed scale, while shaping end-to-end ML software from model design to deployment. #J-18808-Ljbffr Google

Vacancy posted 2 days ago
Similar jobs that could be interesting for youBased on the Senior ML Compiler Engineer for TPUs in San Bruno, CA vacancy
  • Tahoe Therapeutics is seeking a Senior Machine Learning Engineer in South San Francisco. This role involves designing cutting-edge foundation models...  ...responsibilities include adopting advanced techniques in ML and working with scientists to develop compute-optimized models... 
    Senior

    Tahoe Therapeutics

    South San Francisco, CA
    1 day ago
  • Google is looking for a Senior Software Engineer for its AI/ML division focused on YouTube. The ideal candidate will have a strong software development background with expertise in machine learning and technical leadership abilities. This role involves writing and testing... 
    Senior

    Google

    San Bruno, CA
    1 day ago
  • $208k - $263.5k

     ...and improve them until they work at scale. We are roboticists, engineers, operators, and builders. We believe the next great technology...  ...modern deep learning frameworks (PyTorch or JAX) and runtime compilation. Robust programming skills in Python and C++. Familiarity... 
    Senior
    Full time
    Internship
    Work at office
    Flexible hours

    Atoms

    San Francisco, CA
    3 days ago
  • Google Inc. is seeking a Senior Software Engineer specializing in AI/ML for YouTube in San Bruno, California. This role involves developing technologies that change user interaction and requires at least 5 years of programming experience in Python and C++. Candidates should... 
    Senior

    Google Inc.

    San Bruno, CA
    2 days ago
  • A leading biotech company is seeking a Senior Machine Learning Engineer to join its Machine Learning Science team in Brisbane, California. The role requires...  ...and building deep learning pipelines and managing AI/ML systems in the cloud. Candidates should have at least 5... 
    Senior
    Remote job

    Freenome

    Brisbane, CA
    1 day ago
  • $200k - $280k

     ...A leading voice AI startup is seeking a Founding Senior Machine Learning Engineer to fine-tune and deploy human-like voice agents, handling millions...  ...Candidates should have real-world experience in deploying ML models and work well in a fast-paced environment. J-18808-... 
    Senior

    Retell AI

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

    Senior Software Engineer, ML Compilers, TPU, YouTube Note: By applying to this position you will have an opportunity to share your preferred working location...  ...stack. Familiarity with accelerator HW architectures (TPUs/GPUs). About the job Google's software engineers... 
    Senior

    Google

    San Bruno, CA
    2 days ago
  •  ...people who want to push the frontier of what’s possible. As a Senior Machine Learning Engineer, you will play a leading role in designing the next...  ...and an interest in applying cutting‑edge breakthroughs in ML to meaningful problems in drug discovery. We are looking for... 
    Senior
    Work at office
    Visa sponsorship
    Free visa
    3 days per week

    Tahoe Therapeutics

    South San Francisco, CA
    5 days ago
  • An innovative company is seeking a Distributed Systems/ML Engineer to enhance the training throughput of its internal framework. This role involves collaborating with researchers to develop efficient video models and applying cutting-edge techniques to optimize training... 
    Senior

    Jobleads-US

    San Francisco, CA
    5 days ago
  • $161.93k - $247k

     ...Senior Machine Learning Engineer Join to apply for the Senior Machine Learning Engineer role at Freenome. About This Opportunity At Freenome, we are...  ...reliable, scalable artificial intelligence/machine learning (AI/ML) systems in a cloud environment. The MLS team at Freenome... 
    Senior
    Full time
    Work at office
    Local area
    Remote work
    2 days per week
    3 days per week

    Freenome

    Brisbane, CA
    5 days ago
  • An AI-focused technology company in San Francisco is seeking a Senior Machine Learning Engineer to take ownership of the full ML lifecycle. You will design, build, and operate ML systems that directly impact the product, emphasizing reliability and cost-efficiency. Aimed... 
    Senior

    MetAntz

    San Francisco, CA
    1 day ago
  • $138k - $289k

    Indeed is seeking an experienced Machine Learning Engineer located in San Francisco, California. The role involves delivering impactful ML and AI projects, developing and implementing machine learning solutions, and mentoring team members. This position offers a competitive... 
    Senior

    Indeed

    San Francisco, CA
    4 days ago
  • $164k - $312k

    A leading technology company in San Francisco seeks a Senior ML Engineer to drive advanced prediction models for ad ranking systems. You will design and implement efficient systems that optimize engagement predictions. The ideal candidate holds an advanced degree and has... 
    Senior

    Unity

    San Francisco, CA
    4 days ago
  • Arena Club is seeking a Senior Machine Learning Engineer to own and optimize the ranking and discovery systems across its marketplace. You will drive production ML in the request path, delivering measurable lift in conversion and engagement while mentoring junior data scientists... 
    Senior

    Arena Club

    San Francisco, CA
    3 days ago
  • Dormont Manufacturing Co in San Francisco is seeking a skilled individual to optimize model inference and strengthen the core of ComfyUI. Your role will focus on making AI models run faster and more efficiently. The ideal candidate will have experience in production-level...
    Senior

    Dormont Manufacturing Co

    San Francisco, CA
    3 days ago
  • A technology company in San Francisco is seeking an experienced Infrastructure Engineer to ensure 99.99% uptime while working with custom inference stacks and managing GPU loads. The ideal candidate will have deep infrastructure expertise and a passion for tackling complex... 
    Senior

    Morph Inc.

    San Francisco, CA
    1 day ago
  • DoorDash is looking for a Senior Staff Machine Learning Engineer to set the personalization strategy for the consumer shopping journey. In this role, you will implement machine learning solutions that enhance the consumer search experience across various services. Your... 
    Senior

    DoorDash

    San Francisco, CA
    1 day ago
  • Perplexity AI Inc. is seeking experienced ML Engineers to design and optimize core recommendation systems, enhancing user experience. Successful candidates will have hands-on experience in production environments developing personalization, ranking, and user modeling systems... 
    Senior

    Perplexity AI Inc.

    San Francisco, CA
    2 days ago
  • An innovative AI company in San Francisco is seeking a Senior Machine Learning Engineer for Voice AI to optimize voice model inference and productionize...  .... The ideal candidate will have over 5 years of ML engineering experience, strong Python and PyTorch skills,... 
    Senior

    Together AI

    San Francisco, CA
    1 day ago
  • Waymo is seeking a Senior Machine Learning Infrastructure Engineer to lead the development of AI/ML infrastructure for advanced simulations. Candidates should have significant experience in software engineering and ML infrastructure, focusing on large models. This role... 
    Senior

    Waymo

    San Francisco, CA
    1 day ago
  • $213k - $263k

    Waymo in San Francisco is looking for a Machine Learning Engineer to pioneer evaluation systems for its autonomous simulation technology. The role focuses on developing and deploying innovative evaluation approaches to assess realism in multimodal world models. The ideal... 
    Senior

    Waymo

    San Francisco, CA
    1 day ago
  • $183.7k - $248.6k

    Unity is seeking a Senior Machine Learning Infrastructure Engineer in San Francisco to join our Vector Ads team. You will design, build, and maintain the infrastructure that serves ML models in real-time, ensuring they run reliably at scale. Ideal candidates have experience... 
    Senior

    Unity

    San Francisco, CA
    1 day ago
  • Neura Market is looking for experienced ML Engineers to design, build, and optimize recommendation systems that enhance user experiences. The role involves owning the personalization and ranking behind key products to improve user engagement. The ideal candidate will have... 
    Senior

    Neura Market

    San Francisco, CA
    1 day ago
  • $160k - $240k

    Tensec is looking for a Machine Learning Engineer in San Francisco to build algorithms and optimization systems that drive their autonomous decision engine. This role involves designing trading strategies, building execution layers, and deploying robust models that enhance... 
    Senior
    Relocation package

    Tensec

    San Francisco, CA
    1 day ago
  • $155.58k - $320.32k

    Pinterest is seeking a talented engineer skilled in Python to write production code that powers real-time bidding and model training. Ideal...  ...responsibility, including the training and deployment of ML models that drive ad decisions. Located in San Francisco, California... 
    Senior

    Pinterest

    San Francisco, CA
    1 day ago
  • Edison Smart® is seeking a Senior Recommendation & Growth Machine Learning Engineer in San Francisco, CA. You'll design and develop advanced recommendation systems...  .... Join a high-performing team to solve complex ML challenges and enhance personalization across various... 
    Senior

    Edison Smart®

    San Francisco, CA
    1 day ago
  • DoorDash USA is seeking a Senior Staff Engineer in San Francisco to lead the technical direction for AI-first experiences. The role focuses on building modern AI models that enhance ad relevance. Candidates should have over 5 years of experience in machine learning, deep... 
    Senior

    DoorDash USA

    San Francisco, CA
    1 day ago
  • Gravity Engineering Services Pvt Ltd. is seeking a Senior ML Engineer to optimize the model serving layer for voice workloads. This role involves working hands-on with inference engines to enhance model performance for speech applications. The ideal candidate will have... 
    Senior

    Gravity Engineering Services Pvt Ltd.

    San Francisco, CA
    1 day ago
  • $190k - $265k

    Harvey, Inc in San Francisco is looking for a Research Engineer to enhance their AI platform for law, tax, and finance. This role will involve...  ...documents. The ideal candidate will have over 5 years of ML engineering experience and proven ability in advanced research.... 
    Senior

    Harvey

    San Francisco, CA
    1 day ago
  • DoorDash is seeking a Senior Machine Learning Scientist to enhance the search and personalization experiences for our grocery and retail delivery services. You will work on innovative machine learning solutions to improve consumer experiences while mentoring junior members... 
    Senior
    Work at office
    Remote work

    DoorDash

    San Francisco, CA
    1 day ago

Do you want to receive more vacancies?

Subscribe and receive similar vacancies to Senior ML Compiler Engineer for TPUs. Be the first to apply!