Staff Machine Learning Engineer

Arm Holdings

Confirmed live today High trust
Hybrid

Quick summary

Work type
Hybrid
Location
Austin, TX
Salary
$198,100–$268,000 / yr
Posted
71 days ago
Freshness
Confirmed live today

Market check

Salary context

Competitive pay

How this pay compares to similar roles

Similar $230k
This role $233k
$173k $294k
below market most similar roles pay here above market

This role pays less than 55% of similar roles. Most pay $206,175–$254,750 — the blue band above. At the midpoint, this role pays about $233k versus about $230k for comparable roles.

Based on 240 similar postings.

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 113 open roles on FindRole.

Listed pay typically runs $208,450–$282,050 across 112 roles with salary data.

Most-posted roles

View all roles at Arm Holdings

At a glance

TL;DR · Staff Machine Learning Engineer

Staff Machine Learning Engineer The Staff Machine Learning Engineer joins the Content and Algorithm team within the Central Technology ML group to pioneer the technology roadmap for machine learning computing on Arm architecture. This role involves conducting in-depth analyses of complex neural networks on Arm computing platforms to devise algorithms that improve performance, power, and area. You will perform architecture exploration through joint development of hardware, software, and algorithms, specifically focusing on algorithm innovation and prototyping to validate architectural features. Key responsibilities include developing hardware-aware optimization algorithms for state-of-the-art networks on CPU, GPU, and NPU platforms. Required skills include proficiency in Python, computer architecture, and deep learning libraries like Tensorflow and Pytorch. Candidates must have experience training large models on GPU-based systems and knowledge of hardware and software design.

What does a Machine Learning Engineer earn?

Median $235750 from 323 postings across 48 companies.

See salary data

What you'll do

  • Analyze current and future ML workloads on Arm compute platforms with a focus on performance, power, and area.
  • Conduct in-depth analyses of complex neural networks on Arm computing platforms.
  • Innovate and prototype new algorithms to drive and validate architectural features.
  • Develop hardware-aware optimization algorithms for state-of-the-art networks on Arm CPUs, GPUs, and NPUs.
  • Provide ML algorithm expertise to guide joint hardware, software, and algorithm development.
  • Train large deep learning models on powerful GPU-based systems.
  • Develop and work with large software systems using Python.

What we're looking for

  • 3+ years of experience in ML Algorithm Development and/or Optimization environments.
  • Master or PhD degree in Computer Engineering, Electrical Engineering, Computer Science, or a related technical field.
  • Proficiency in computer architecture with basic knowledge of hardware and software design.
  • Experience developing and working with large software systems using programming languages like Python.
  • Knowledge of innovative deep learning libraries such as TensorFlow and PyTorch.
  • Experience training large deep learning models on powerful GPU-based systems.
  • Knowledge of ML model optimization techniques for resource-constrained ARM edge compute platforms (preferred).

More like this

Similar roles

Staff Machine Learning Software Engineer

Qualcomm

San Diego, CA 40 days ago $158,400–$237,600
C C++ Python TensorFlow PyTorch ARM DSP GPU ML Kernels Compilers Embedded Software Performance Optimization
3+ yrs exp

Senior Machine Learning Software Engineer

Qualcomm

San Diego, CA 40 days ago $140,800–$211,200
C C++ Python TensorFlow PyTorch ARM DSP GPU ML Kernels Compilers Embedded Software Performance Optimization Verification IoT Machine Learning

Staff Machine Learning Engineer

Capital One Financial

Plano, TX +1 47 days ago
Machine Learning Python C++ Scala C Dask RAPIDS NumPy Pandas Scikit-learn Distributed Computing Data Pipelines High Performance Computing
10+ yrs exp

Staff Machine Learning Engineer

Intuit

Mountain View, CA 64 days ago $230,000–$250,000
Python PyTorch TensorFlow Pandas NumPy LLM RAG LangChain AWS SageMaker Kubernetes JavaScript Java Microservices NoSQL System Design Neural Networks
8+ yrs exp

Staff Machine Learning Engineer

Adobe

San Jose, CA +1 34 days ago
Generative AI Agentic AI Computer Vision Multimodal Learning LLMs Transformers Diffusion Models Vision-Language Models (VLMs) Python PyTorch AWS Azure Docker Kubernetes Model Context Protocol (MCP) APIs
5+ yrs exp

Staff Machine Learning Engineer

Capital One Financial

Plano, TX +1 28 days ago $269,100–$307,200
Python Java GoLang C++ PyTorch TensorFlow Spark Ray Kubernetes AWS GCP Azure Dask RAPIDS Pandas NumPy Scikit-learn Transformers RNNs CNNs LSTMs Data Pipelines