Job Description

Our client has developed the next-generation Tensor Processing Units (TPUs), custom-designed AI accelerators optimised for training and inference of large AI models. With working prototypes already, they are now looking to scale the business as quickly as possible.

We’re seeking a highly experienced and motivated individual to design and build the software architecture for our next-generation GPUs. This role demands deep expertise in C and C++ programming, low-level programming, compiler construction, and optimisation techniques.

Role: Staff Software Engineer

Location: London

Salary: DOE / Competitive + benefits

Responsibilities:

• Design and develop the software architecture for the next-generation TPU, ensuring high performance and scalability.
• Collaborate with hardware engineers to integrate software and hardware components seamlessly.
• Optimise software performance through advanced techniques in low-level programming and compiler design.
• Develop and maintain machine learning frameworks and tools to leverage the full potential of the TPU.
• Conduct code reviews, provide technical mentorship, and guide other team members in best practices.
• Stay current with industry trends and advancements in GPU technologies, machine learning, and optical computing.
• Lead and participate in the development of technical documentation and specifications.
• Drive innovation and contribute to the strategic direction of the software engineering team.

Skills and Experience:

• 8+ years of experience in software engineering with a focus on C and C++ programming.
• Extensive experience in compilers, low-level programming, and optimisation techniques.
• Proven expertise in machine learning and its applications in high-performance computing.
• Strong problem-solving skills and the ability to think critically and creatively.
• Experience in high-paced, dynamic work environments.
• Bachelor's degree in computer science, electrical engineering, telecoms engineering, mathematics, or a related field.
• Excellent teamwork and communication skills, with the ability to collaborate effectively with cross-functional teams.
• Personal projects are a key differentiating factor and hold more weight than other requirements.