Senior ML Compiler Engineer, GPU

Requirements:

• Bachelor's degree in Computer Science, Information Technology, Engineering, a related technical field, or equivalent practical experience.
• 5 years of experience in software development.
• Experience in programming languages such as C++.

Nice to haves:

• Experience in contributing to an open-source project.
• Experience working in HPC/GPU workloads.
• Experience with compilers, compiler construction, and Low-Level Virtual Machine (LLVM).
• Experience with ML frameworks (e.g., PyTorch, JAX, TensorFlow).
• Experience in computer science, with strong expertise in data structures, algorithms, and software design.
• Ability to communicate and navigate within rapidly changing environments.

What you'll be doing:

• Partner with users of XLA on GPU as part of a high-investment and high-priority program to make ML on GPU successful at Google.
• Design and launch new GPU optimizations and features, including performance and correctness testing.
• Enhance the GPU compiler software stack so that it performs faster and more efficiently on ML workloads.
• Foster deep collaboration between the GPU Software team, customer teams, Google research and the open-source community.

Perks and benefits:

• Google Cloud's software engineers develop the next-generation technologies.
• Opportunities to switch teams and projects.
• Anticipate customer needs and act like an owner.
• Versatile engineers, display leadership qualities and be enthusiastic to take on new problems.