Google is seeking to improve generalization across different hardware, frameworks, and use cases by simplifying and standardizing compiler integration across different stacks for the XLA compiler, which is used in the Research to Production pipeline for Training and Serving use cases on TPU, GPU, and CPU accelerators.
Requirements
- Bachelor’s degree or equivalent practical experience.
- 2 years of experience with software development in C++.
- 2 years of experience with developing large-scale infrastructure, distributed systems or networks, or experience with compute technologies, storage or hardware architecture.
- 2 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture.
- Master's degree or PhD in Computer Science or a related technical field.
- 2 years of experience with performance, large-scale systems data analysis, visualization tools, or debugging.
- Experience in compilers or runtimes.
Responsibilities
- Write and test product or system development code.
- Understand how accelerator compilers and runtimes interact at a high level.
- Develop and apply metrics to understand the problem you are solving and gage status/success as needed.
- Close infrastructure (infra) gaps to help with ML stack maturation (e.g., reduce a number of ways something is done, improve reproducibility, improve tooling, improve usability).
- Participate in design reviews with peers and stakeholders to decide amongst available technologies.
Other
- Bachelor’s degree or equivalent practical experience.
- Master's degree or PhD in Computer Science or a related technical field.
- Ability to work in the US.
- Proficiency in code and system health, diagnosis and resolution, and software test engineering.
