Charterhouse is working with an innovative technology startup specializing in the design and development of advanced, customized server hardware solutions. The company has a big focus on ML (Machine Learning) applications and AI (Artificial Intelligence). They are growing and looking to hire an AI Performance Engineer.
About the role
The AI Performance Engineer should be able to evaluate and improve the performance of AI workloads across various hardware platforms and machine learning frameworks. This role will work on optimizing AI systems and ensuring they meet performance goals across a wide range of hardware, including CPUs, GPUs, and custom ASICs.
The AI Performance Engineer will also be responsible for profiling and enhancing the performance of AI workloads, identifying appropriate benchmarks, and analysing performance across different AI algorithms and models. Additionally, this role will collaborate closely with software and firmware teams to ensure smooth integration and optimal performance from end to end.
About you
The successful candidate will bring at least 5 years of experience in C++ and Python programming, along with an in-depth understanding of CPU and GPU architectures, as well as custom ASICs like NPUs and TPUs. The individual should be skilled in low-level optimization techniques, memory hierarchies, and understanding performance trade-offs in high-performance computing settings.
Experience with the internals of deep learning frameworks such as PyTorch or TensorFlow, and a solid understanding of deep learning models, is essential. The ideal candidate should be comfortable with the training and deployment of machine learning models, and have experience with distributed systems and parallel programming. A strong grasp of software development best practices, including testing, profiling, debugging, and version control, is vital for success in this position. Excellent communication skills in English are essential for this position.
