Software Engineer III, OpenXLA

Google is a leading technology company that develops innovative products and services used by billions of users worldwide. Known for its search engine, Google has expanded into various areas including cloud computing, artificial intelligence, and hardware.
$136,000 - $200,000
Backend
Senior Software Engineer
In-Person
2+ years of experience
AI

Description For Software Engineer III, OpenXLA

Google is seeking a Software Engineer III for their OpenXLA team. OpenXLA is an open-source ML compiler that powers Tensorflow, JAX and PyTorch/XLA, accelerating ML models on CPUs and GPUs. The role involves working on the hardware-independent parts of the stack, including StableHLO, HLO, XLA components and APIs, and compiler tooling for faster model debugging. The ideal candidate will contribute to making OpenXLA the best platform for ML acceleration innovation and ensure it's accessible for hardware vendors and compiler researchers. This position is part of Google's Core team, which builds the technical foundation behind Google's flagship products. The role offers opportunities to impact important technical decisions across the company and contribute to central solutions that break down technical barriers and strengthen existing systems. The position requires a strong background in software development, data structures, algorithms, and machine learning, with a focus on compiler development and open source experience.

Last updated 2 months ago

Responsibilities For Software Engineer III, OpenXLA

  • Write product or system development code
  • Participate in, or lead design reviews with peers and stakeholders to decide among available technologies
  • Review code developed by other developers and provide feedback to ensure best practices
  • Contribute to existing documentation or educational content and adapt content based on product/program updates and user feedback
  • Triage product or system issues and debug/track/resolve by analyzing the sources of issues and the impact on hardware, network, or service operations and quality
  • Contribute to the teams mission by learning and contributing to StableHLO, XLA and help in engaging with stakeholders who integrate with OpenXLA

Requirements For Software Engineer III, OpenXLA

Python
  • Bachelor's degree or equivalent practical experience
  • 2 years of experience with software development in one or more programming languages, or 1 year of experience with an advanced degree in an industry setting
  • 2 years of experience with data structures or algorithms in either an academic or industry setting
  • 2 years of experience with Machine Learning (ML) algorithms and tools (e.g., TensorFlow), Artificial Intelligence (AI), deep learning or natural language processing

Benefits For Software Engineer III, OpenXLA

  • bonus
  • equity
  • benefits

Interested in this job?

Jobs Related To Google Software Engineer III, OpenXLA

Senior Customer and Partner Solutions Engineer, gTech

Senior Customer and Partner Solutions Engineer position at Google, focusing on technical problem-solving and partner integration support with competitive compensation and benefits.

Senior Software Engineer, P2020 Rollouts

Senior Software Engineer position at Google working on P2020 Rollouts, focusing on cloud infrastructure and service management optimization.

Senior Software Engineer, Google Research

Senior Software Engineer position at Google Research, focusing on developing next-generation technologies and conducting fundamental research that impacts billions of users.

Senior Optical Transport Network Engineer

Senior Optical Transport Network Engineer role at Google, focusing on network infrastructure and optimization.

Senior Software Engineer, Ads

Senior Software Engineer position at Google Ads team in Warsaw, focusing on full-stack development with 5+ years of experience required.