Join the Central Technology ML (CTML) team as a key member of the Content and Algorithm group within the Center of Excellence for Machine Learning, where you will drive joint development initiatives by conducting in-depth analyses of complex neural networks on ARM's computing platforms. Your daily tasks include developing hardware-aware optimization algorithms to enhance state-of-the-art networks across CPU/GPU/NPUs, focusing on performance, power, and area (PPA) metrics. Ideal candidates have 3+ years of experience in ML algorithm development or optimization, a Master’s or PhD degree in relevant fields, proficiency in Python, and expertise with deep learning libraries like TensorFlow and PyTorch. This role demands knowledge of large software systems and the ability to train models on powerful GPU-based systems, all while contributing to cutting-edge advancements in machine learning technology for ARM architecture.
Skills
Python
TensorFlow
PyTorch
GPU
ARM
CPU
NPU
Deep Learning
ML Model Optimization
Computer Architecture
CI/CD
What you'll do
Conduct in-depth analyses of complex neural networks on ARM compute platforms.
Develop hardware-aware optimization algorithms for state-of-the-art networks on Arm platforms.
Innovate and prototype ML algorithms to drive architectural feature validation.
Optimize ML models for resource-constrained ARM edge computing platforms.
Collaborate on joint development initiatives across hardware, software, and algorithms.
What we're looking for
3+ years of experience in ML algorithm development and optimization.
Master’s or PhD degree in Computer Engineering, Electrical Engineering, or related fields.
Proficiency in Python for developing large software systems.
Expertise in deep learning libraries like TensorFlow and PyTorch.
Experience training large deep learning models on GPU-based systems.
Knowledge of hardware-aware ML model optimization techniques.
Understanding of ARM compute platforms’ performance, power, and area constraints.
Market check
Salary context
This
$249,900–$338,100
range sits above
95%
of similar postings on FindRole.
Peer median band
$160,500–$260,000
Median floor and ceiling across peers.
Typical midpoint (25–75%)
$176,000–$246,150
Middle half of comparable postings.
Based on 240 comparable postings.*
* 240 is the maximum number of comparable postings sampled.
Employer
About Arm Holdings
Arm Holdings plc is a leading British semiconductor and software design firm, established in 1990 and recognized for developing energy-efficient processor architectures that power nearly all smartphones and a vast range of IoT and computing devices.
Arm Holdings currently has
34 open roles
on FindRole.
Listed pay typically runs
$184,500–$249,600
across 34 roles with salary data.