Description
Google Cloud is hiring a Software Engineer, TPU Compiler Optimization to build infrastructure for machine learning and drive TPU performance. The role involves researching and developing compiler optimizations, improving ML workload efficiency through performance debugging and custom kernels, integrating frameworks such as PyTorch and JAX, and providing technical leadership and mentorship as a Team Lead. Candidates need a bachelor's degree or equivalent practical experience, 8 years of software development experience, 3 years of software design and architecture experience, and experience with machine learning, compilers, computer architecture, GPU programming, C++, and Python.
