PhD Software Engineer, TPU Compiler & AI Optimizations

WeAreTechWomen; United Kingdom; On-site; GBP 65,000 - 80,000; Full time

Job summary

A leading tech company in the United Kingdom is seeking a Software Engineer to work on the TPU compiler and optimize machine learning models. Candidates should have strong backgrounds in data structures, artificial intelligence, and parallel computing, and preferably be pursuing or hold a PhD. As part of the fast-paced environment, you will write code in C++ and lead collaborations in design reviews. This role offers a chance to work on impactful technologies and contribute to cutting-edge solutions.

Qualifications

• Experience with coding in data structures, algorithms, and software design.
• Research experience in Artificial Intelligence, Distributed Systems, and related fields.
• Currently enrolled in or graduated from a PhD program.
• Experience working with parallel computing and compiler construction.

Responsibilities

• Write product or system development code for the TPU compiler in C++.
• Participate in design reviews with peers and stakeholders.
• Contribute to a compiler which scales machine learning models across accelerators.
• Design and implement performance optimizations for critical features.

Skills

Coding in data structures

Algorithms

Software design

Artificial Intelligence

Machine Learning

Debugging concurrent computations

Experience with GPUs

Education

PhD program

Tools

C++

Job description

A leading tech company in the United Kingdom is seeking a Software Engineer to work on the TPU compiler and optimize machine learning models. Candidates should have strong backgrounds in data structures, artificial intelligence, and parallel computing, and preferably be pursuing or hold a PhD. As part of the fast-paced environment, you will write code in C++ and lead collaborations in design reviews. This role offers a chance to work on impactful technologies and contribute to cutting-edge solutions.