Principal GPU Architect

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 :

Nice to Haves :

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 accommodations@arm.com. Salary range : $241,100 - $326,100 per year. We offer a flexible hybrid working environment tailored to team needs.

#J-18808-Ljbffr