Staff Software Engineer, AI/ML Performance and Accelerators

Job description

Minimum qualifications: - Bachelor’s degree or equivalent practical experience. - 8 years of experience with software development in C++ and Python. - 5 years of experience with ML infrastructure, including performance analysis and optimisation of distributed systems. - 5 years of experience modeling or evaluating performance for large-scale machine learning workloads (e.g., Large Language Models). - Experience with hardware-software co-design, ML accelerators, and low-level software stacks such as compilers. Preferred qualifications: - Master’s degree or PhD in Computer Science, or a related technical field. - Experience in the self-driving or automotive industry building large-scale distributed infrastructure and networking systems. - Experience in building high-fidelity HW/SW simulators (C++/Python) and large-scale performance tooling for ML accelerators. - Understanding of large-scale AI workloads (LLMs), agentic systems, and performance/cost analysis for hyperscale systems. - Knowledge of compiler flows, hardware-software co-design, and translating frameworks like PyTorch or JAX to hardware instructions. - Ability to collaborate cross-functionally to translate performance insi...