Senior Principal ML Compiler Engineer

Key Skills

Algorithms, AWS, Azure, C++, Machine learning techniques, Python Programming, PyTorch, TensorFlow

Job Description

Job 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 to reward people competitively 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

Accommodations at Arm

At Arm, we want our people to Do Great Things. If you need support or an accommodation to Be Your Brilliant Self during the recruitment process, please email accommodations@arm.com. To note, by sending us the requested information, you consent to its use by Arm to arrange for appropriate accommodations. All accommodation requests will be treated with confidentiality, and information concerning these requests will only be disclosed as necessary to provide the accommodation. Although this is not an exhaustive list, examples of support include breaks between interviews, having documents read aloud or office accessibility. Please email us about anything we can do to accommodate you during the recruitment process.

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. Within that framework, we empower groups and teams to determine their own particular hybrid working pattern, depending on the work and the team’s needs. Details of what this means for each role will be shared upon application. 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. Please talk to us to find out more about what this could look like for you.