Job Details

Job Description:

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.

As a member of the CTML Content and Algorithm team, the candidate will play a key role within the Center of Excellence for Machine Learning. They will actively contribute to this joint development initiative, conducting in-depth analyses of complex neural networks on ARM's computing platforms and devising algorithms that will elevate us to the forefront of the future.

Responsibilities:

This exciting position offers the opportunity to work on the latest technology in a collaborative environment to:

• Develop an in-depth understanding of current and future ML workloads on ARM compute platforms, focusing on performance, power, and area (PPA).
• Engage in architecture exploration that requires the joint development of hardware, software, and algorithms. This role acts as the ML Algorithm authority within this exploration process, including algorithm innovation and prototyping to drive and validate architectural features.
• Develop hardware-aware optimization algorithms that can enhance state-of-the-art networks on Arm platforms, including CPU, GPU, and NPUs.

Required Skills and Experience:

• 3+ years of experience in ML Algorithm Development and/or Optimization.
• A Master's or PhD degree in Computer Engineering, Electrical Engineering, Computer Science, or related fields.
• Proficiency in computer architecture and basic knowledge of hardware and software design.
• Experience developing and working with large software systems in programming languages like Python.
• Knowledge of deep learning libraries such as TensorFlow and PyTorch.
• Experience training large deep learning models on GPU-based systems.

Nice to Haves:

• ML model optimization techniques targeted for resource-constrained ARM edge compute platforms.

Additional Information:

At Arm, we value our people and aim to contribute to advancing ML performance across the industry. We support diversity and equal opportunity in our workplace. If you require accommodations during the recruitment process, please contact ...@arm.com. Salary range: $241,100 - $326,100 per year. We offer a flexible hybrid working environment tailored to team needs.

J-18808-Ljbffr