Machine Learning Architect

Job Overview: ----------------- High-performance ML workloads on Arm CPUs requires the co-development of algorithms and highly optimized CPU kernels. In CT-ML (Central Technology, Machine Learning), rapid kernel prototyping is crucial for exploring algorithms and assessing trade-offs between model accuracy and performance. Successful prototypes are essential to drive future CPU architecture development and also deliverables to Central Engineering for final production. Responsibilities: --------------------- This position is part of a dedicated team within the CT-ML group to focus on analyzing ML workload, rapid prototyping of highly optimized CPU kernels to drive model performance and accuracies. Required Skills and Experience : ------------------------------------ * Strong interest and passion for implementing high-performance kernel code in a dynamic environment. * 4+ years experience in implementing high performance CPU kernel with vector and matrix extensions. * Experience measuring and understanding performance * Experience in creating an efficient kernel code development framework including tools and testing * Deep understanding on CPU architecture “Nice To Have” Skills and Experience : ------------------------------------------ * Knowledge of ML models and algorithm is a plus * Advanced degree or equivalent experience in Computer Architecture and Software are a plus In Return: -------------- Arm is committed to global talent acquisition, offering an attractive relocation package. With offices around the world, Arm is a diverse organization of dedicated, creative and highly talented engineers. By enabling a dynamic, inclusive, meritocratic, and open workplace, where all our people can grow and succeed, we encourage our people to share their unrivaled contributions to Arm's success in the global marketplace.