Senior Software Engineer, Performance, SOC Architect

Job description

Write and test product or system development code. Learn and build an intuitive understanding of existing data collection, analysis, and visualization workflows. Support new and exciting ML paradigms (such as horizontal scaling for upcoming TPU chips) by making contributions across the end-to-end stack and analysis tools. Partner with Product Area leads to understand model optimization use cases, drive cross functional efforts to deliver on chip profiling requirements, and propose new hardware features. Collaborate across Hardware, Driver, Runtime, and Performance Analysis teams and many other stakeholders. Minimum Qualifications: Bachelor’s degree or equivalent practical experience. 5 years of experience with software development in one or more programming languages. 3 years of experience with performance, large-scale systems data analysis, visualization tools, or debugging. 3 years of experience with Computer Architecture, C++, Python, CUDA, GPU Programming, GPU Drivers, System On a Chip, Application-Specific Integrated Circuit. 3 years of experience testing, maintaining, or launching software products, and 1 year of experience with software design and architecture. Preferred Qualifications: Master's degree or PhD in Computer Science or related technical field. 5 years of experience with data structures and algorithms. 1 year of experience in a technical leadership role. Experience with GPU or TPU performance analysis, and a passion for developer productivity. Experience with ML frameworks such as TensorFlow, JAX, and PyTorch, or ML compilers such XLA. Proven track record in open-source software development, including experience in releasing and supporting open-source projects.