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.
* Bachelors 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.
J-18808-Ljbffr