ARM Principal Machine Learning Performance Kernel Engineer 

United Kingdom, England, Cambridge 

103996446

Develop in depth understanding of ML workload, develop algorithms and optimization techniques to drive PPA (Performance, Power, Area) on current and future Arm platforms.

1. Implementation of performance critical machines specific kernel or ARM assembly code on dedicated hardware accelerator and CPU to drive algorithm and architecture exploration
2. Collaborate with ML Algorithm development team to explore performance critical analysis
3. Develop internal tooling capabilities to support algorithmic and architecture exploration

• Develop performance critical assembly code to accelerate ML operations done in low precision arithmetic
• Develop performance critical kernel to accelerate fusing of multiple ML operations on accelerators
• Explore different quantization schemes and assess performance PPA trade off

• 3+ years experience in developing performance critical kernel on dedicated accelerator, GPU or CPU
• Graduate students in Computer Engineering, Electrical Engineering, Computer Science or other related technical fields
• Deep knowledge in machine learning, deep learning, and neural network design, optimization, and compression techniques
• Overall high-level knowledge of computer architecture, systems, and HW-SW co-design
• Ability to develop and work with large software systems in programming languages like Python
• Knowledge of cutting-edge deep learning libraries such as Tensorflow, and Pytorch
• Willing to learn and train large deep learning models on GPU-based systems

• Experience with ML model design, optimization, and HW-SW co-development methodology
• ML Model Optimization techniques targeting PPA (Performance, Power, and Area) of neural networks on ARM compute platforms
• Adaptability to the fast-moving ML industry and willingness to learn new technology in a very dynamic environment

Working on interesting new projects with leaders in the field is exciting, but we also know how important it is to receive support. That's why throughout your internship, you can expect regular feedback and development opportunities, social activities to connect with your peers, an end of internship celebration, plus the opportunity to be *considered for future Graduate positions (*subject to performance). #getreadytogrow

In addition to a competitive salary and rewards package, our on-the-job learning and mentoring/buddy schemes provide unparalleled learning and networking opportunities from the best in the industry.