Software Engineer, ML Accelerators, DeepMind

Job description

Direct full-stack Software (SW) role, focusing on ML compiler and inference infrastructure. Design and develop ML compiler stack to bridge AI workloads and low-level Hardware (HW) operators. Design and develop SW infrastructure to enable high performance inference. Drive SW/HW codesign, kernel optimization, performance tuning, and debugging. Minimum Qualifications: Bachelor's degree in Computer Science, a related technical field, or equivalent practical experience. Experience in software development using Python and C++. Experience with machine learning (ML) hardware accelerators, ML compiler stacks (e.g., JAX, PyTorch, XLA), and kernel development. Preferred Qualifications: PhD degree in Computer Engineering, Computer Science, or a related field. 2 years of experience with full-stack system development for hardware accelerators. 2 years of experience with collaborative, hyper-scalar software development and engineering best practices. Experience scoping, planning, and executing projects towards team goals in ambiguous environments.