Arm

Staff Machine Learning Engineer

Arm
Apply
21 days ago

Base Salary

$250k - $338k/yr

Responsibilities

  • Analyze current and future machine-learning workloads on Arm computing platforms with emphasis on performance, power, and area.
  • Serve as the ML algorithm expert in joint hardware, software, and algorithm architecture exploration.
  • Develop, innovate, and prototype algorithms to drive and validate architectural features.
  • Develop hardware-aware optimization algorithms for state-of-the-art networks on Arm CPUs, GPUs, and NPUs.

Requirements

  • At least 3 years of experience in ML algorithm development and/or optimization.
  • Master's or PhD degree in Computer Engineering, Electrical Engineering, Computer Science, or a related technical field.
  • Proficiency in computer architecture and basic knowledge of hardware and software design.
  • Experience developing and working with large software systems using programming languages such as Python.
  • Knowledge of deep-learning libraries including TensorFlow and PyTorch.
  • Experience training large deep-learning models on GPU-based systems.
  • Preferred: experience with ML model optimization techniques for resource-constrained Arm edge-computing platforms.

Benefits

  • Hybrid working arrangements are determined by teams, with role-specific details provided during the application process.
  • The company provides support and accommodations during recruitment, including interview breaks, documents read aloud, and office accessibility support.
  • Arm offers competitive total rewards, with the total reward package shared during the recruitment process.

Tech Stack

PythonPyTorchTensorFlow

Categories

Arm

About Arm

5,001-10,000 employees

Arm’s foundational technology is defining the future of computing. A future built by the greatest technology ecosystem in the world. A future built on Arm. Arm is everywhere technology matters. Technology matters everywhere. Together, we’ll power every technology revolution moving forward, including cloud computing, automotive and autonomous systems, IoT, the metaverse, and beyond. Changing the world. Again. On Arm.

Contact me