Staff ML Engineer — Hardware‑Aware ML on ARM Architectures

Arm; Austin (TX); Hybrid; USD 250,000 - 338,000; Full time

Job summary

Arm is seeking a skilled ML Algorithm Developer in Austin to push the frontier of machine learning on Arm-based hardware. You will explore architecture, prototype algorithms, and optimize performance across CPU/GPU/NPUs while collaborating across hardware, software, and ML teams.

The role emphasizes advancing deep learning models on Arm systems, with strong experience in Python and major DL libraries such as TensorFlow and PyTorch. Hybrid working arrangements may apply in this location.

Qualifications

• 3+ years in ML algorithm development/optimization.
• Proficient in Python for large software systems.
• Strong knowledge of computer architecture and HW/SW design.

Responsibilities

• Drive architecture exploration with hardware, software and algorithm co-design; contribute as ML algorithm expert.
• Develop hardware-aware optimization algorithms to run state-of-the-art networks on Arm CPUs/GPUs/NPUs.

Skills

ML Algorithm Development

Python

Computer Architecture

TensorFlow

PyTorch

GPU Training

Model Optimization

Education

Master's or PhD in Computer Engineering / Electrical Engineering / Computer Science

Job description

Arm is seeking a skilled ML Algorithm Developer in Austin to push the frontier of machine learning on Arm-based hardware. You will explore architecture, prototype algorithms, and optimize performance across CPU/GPU/NPUs while collaborating across hardware, software, and ML teams.

The role emphasizes advancing deep learning models on Arm systems, with strong experience in Python and major DL libraries such as TensorFlow and PyTorch. Hybrid working arrangements may apply in this location.