GlossGenius is seeking a Staff Software Engineer to join the Machine Learning Compiler (MPACT) team in Google Research. You will focus on designing continuous integration systems and developing compiler tools for Google's TPU accelerators, enabling high-performance computing development.
What You'll Do
- Help design and develop a continuous integration and continuous delivery system that manages the building and testing of software across open source and Google infrastructures.
- Help build LLVM/MLIR-based C/C++/Cuda compiler and simulation tools for the Google TPU family of ML accelerators.
What We're Looking For
- Bachelor's degree in Computer Science or equivalent practical experience.
- 8 years of experience with software development in one or more programming languages (e.g., Python, C, C++, Java, JavaScript).
- 5 years of experience with compiler development.
- Experience with LLVM development, modifying LLVM as a contributor.
Nice to Have
- Master's degree in Computer Science or equivalent practical experience.
- Experience with software package management, building packages from RedHat and Debian for deployment into artifact registries.
- Experience developing Machine Learning models in JAX, TensorFlow frameworks, or equivalent.
- Experience with large-scale systems (e.g., HPC) design and management.
- Experience with MLIR, including design and implementation of MLIR-based compilers.
Technical Stack
- Languages: Python, C, C++, Java, JavaScript
- Compiler/Tools: LLVM, MLIR, Cuda
- ML Frameworks: JAX, TensorFlow
Team & Environment
You will join the Machine Learning Compiler (MPACT) team within Google Research.
Benefits & Compensation
- Compensation: $207,000-$300,000 + equity
- Health, dental, vision, life, and disability insurance
- Retirement: 401(k) with company match
- Paid Time Off: 20 days of vacation per year, accruing at 6.15 hours per pay period for the first five years
- Sick Time: 40 hours per year (statutory); 5 days per event (discretionary)
- Maternity Leave (Short-Term Disability + Baby Bonding): 28-30 weeks
- Baby Bonding Leave: 18 weeks
- Holidays: 13 paid days per year
Work Mode
This position is based in one of our local offices: Austin, TX; Kirkland, WA; Mountain View, CA; New York, NY; or Sunnyvale, CA.
Google is proud to be an equal opportunity workplace and an affirmative action employer. We are committed to equal employment opportunity regardless of race, color, ancestry, religion, sex, national origin, sexual orientation, age, citizenship, marital status, disability, gender identity, or Veteran status.
