Staff Machine Learning Engineer

What you'll do

• Analyze complex neural networks on Arm computing platforms to evaluate performance, power, and area.
• Develop hardware-aware optimization algorithms for state-of-the-art networks on CPU, GPU, and NPU architectures.
• Innovate and prototype new machine learning algorithms to drive and validate architectural features.
• Conduct in-depth research into future ML workloads specifically tailored for Arm platforms.
• Serve as a subject matter expert for ML algorithms during the joint hardware and software development process.
• Develop and implement large-scale software systems using Python and deep learning libraries like TensorFlow or PyTorch.
• Optimize machine learning models for resource-constrained Arm edge computing devices.

What we're looking for

• 3+ years of experience in an ML Algorithm Development and/or Optimization environment.
• A 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.
• ML Model Optimization techniques targeted for resource-constrained ARM edge compute platforms (preferred).