Principal ML Software Engineer
ARM

Job Info

---

Role Overview

Do you want to help develop next generation machine learning technology for Arm platforms?

Here is an excellent opportunity to work on technology which will be used to drive our future HW and SW roadmaps! We are looking for an experienced engineer who is passionate about developing compilers and loves open-source software.

As a member of the Central Technology Machine Learning team, you will be part of a team working closely with technology counterparts in the hardware and software groups. Your work will involve hands-on invention, prototyping and evaluation of new compilation techniques for enabling a wide range of machine learning models to run on many different implementations.
Key Responsibilities:

• Invent and prototype new machine learning compilation technologies
• Collaborate with internal and external stakeholders in technology and product groups
• Contribute to open-source software projects

Required Skills and Experience:

• Excellent programming skills in C/C++
• Experience with machine learning models from either Tensorflow or Pytorch

"Nice To Have" Skills and Experience:

• Experience developing compiler toolchain(s)
• Experience with MLIR and common MLIR dialects such as linalg, TOSA, etc
• Experience with advanced compilation techniques like auto-parallelization, auto-vectorization, loop transformations, and cache-aware scheduling

In Return:

You will be working on the forefront of Machine Learning technology at Arm. Arm® technology is at the heart of a computing and connectivity revolution that is transforming the way people live and businesses operate. From the unmissable to the invisible; our sophisticated, energy-efficient processor designs are enabling the intelligence in 86 billion silicon chips and securely powering products from the sensor to the smartphone to the supercomputer. With more than 1,000 technology partners including the world's most famous business and consumer brands, we are driving Arm innovation into all areas compute is happening inside the chip, the network and the cloud.

With offices around the world, Arm is full of diverse, dedicated, creative and hard-working engineers. By enabling an inclusive, meritocratic and open workplace where all our people can grow and succeed, we support our people to share their contributions to Arm's success in the global marketplace.

Salary Range:

From: $ 219,500

To: $ 301,900

"We value people as individuals and our commitment is toreward peoplecompetitively and equitably for the work they do and the skills and experience they bring to Arm. Salary is only one component of Arm's offering. The total reward package will be shared with candidates during the recruitment and selection process."

#LI-SA1
Equal Opportunities at Arm

Arm is an equal opportunity employer, committed to providing an environment of mutual respect where equal opportunities are available to all applicants and colleagues. We are a diverse organization of dedicated and innovative individuals, and don't discriminate on the basis of any characteristic.
Hybrid Working at Arm

Arm's hybrid approach to working is centred around flexibility, where we split our time between the office and other locations to get our work done. We empower groups/teams to determine their hybrid working pattern, with an expected minimum attendance in the office of 20% per month, depending on the work, level of experience and team's needs. In some cases, the flexibility we can offer is limited by local legal, regulatory, tax, or other considerations, and where this is the case, we will collaborate with you to find the best solution.

This job has expired.