Senior ML Algorithm Engineer for Edge Compute

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

Benefits offered by this job

Hybrid work model

Accommodation support

Job summary

Arm Limited's Central Technology ML (CTML) team is seeking a ML Algorithm Developer to advance ML workloads on Arm compute platforms, with emphasis on performance, power and area (PPA).

You will contribute to hardware-aware optimization, prototype algorithms, and scalable software systems in Python using TensorFlow and PyTorch, training large models on GPU clusters.

Join a 30+ year history of innovation and help push Arm's ML capabilities to the next frontier.

Qualifications

• 3+ years in ML algorithm development or optimization.
• Master’s or PhD in Computer Engineering, Electrical Engineering, CS, or related field.
• Proficient in computer architecture; HW/SW design knowledge.
• Experience with large software systems in Python.
• Familiar with TensorFlow and PyTorch.
• Experience training large DL models on GPU clusters.

Responsibilities

• Develop deep understanding of ML workloads on Arm compute platforms focusing on PPA.
• Lead ML algorithm innovations and prototyping for architecture features.
• Create hardware-aware optimization algorithms for state-of-the-art networks.

Skills

ML Algorithm Dev

Python

TensorFlow

PyTorch

Computer Architecture

Education

Masters/PhD in CS/EE/CE

Tools

GPU Programming

HW/SW Co-design

Job description

Arm Limited's Central Technology ML (CTML) team is seeking a ML Algorithm Developer to advance ML workloads on Arm compute platforms, with emphasis on performance, power and area (PPA).

You will contribute to hardware-aware optimization, prototype algorithms, and scalable software systems in Python using TensorFlow and PyTorch, training large models on GPU clusters.

Join a 30+ year history of innovation and help push Arm's ML capabilities to the next frontier.