Staff Machine Learning Engineer

Job Overview The mission of the Central Technology ML (CTML) team is to pioneer and implement the technology roadmap that paves the way for the future of machine learning computing on Arm architecture. We believe that innovative ML technology necessitates joint development across hardware, software, and ML algorithms. We will collaborate on what is the best way to balance among all these constraints.

Responsibilities Develop an in‑depth understanding of current and future ML workloads on ARM compute platforms with a focus on performance, power, and area (PPA). Act as the ML algorithm expert within the architecture exploration process, including algorithm innovation and prototyping to drive and validate architectural features. Design hardware‑aware optimization algorithms that enable state‑of‑the‑art neural networks to run efficiently on ARM CPU/GPU/NPUs.

Required Skills and Experience 3+ years of experience in ML algorithm development and/or optimization. Master’s or Ph.D. degree in Computer Engineering, Electrical Engineering, Computer Science, or related technical field. Proficiency in computer architecture and basic knowledge of hardware and software design.

Experience developing and maintaining large software systems in Python. Knowledge of deep‑learning libraries such as TensorFlow and PyTorch. Experience training large deep‑learning models on GPU systems.

Nice to Have Skills and Experience Experience with model‑optimization techniques for resource‑constrained ARM edge compute platforms. Salary Range $249,900‑$338,100 per year Equal Opportunities Arm is an equal‑opportunity employer, committed to providing an environment of mutual respect where equal opportunities are available to all applicants and colleagues. Arm prohibits discrimination or harassment of any kind based on race/ethnicity, religion, national origin, age, sex, sexual orientation, gender identity and expression, disability, pregnancy, medical condition, marital status, citizenship status, military/veteran status, and other characteristics protected by law. #J-18808-Ljbffr