Software Engineer - AI Frameworks

Are you an enthusiastic and motivated engineer who wants to be part of a team building the future of AI on Arm?

Job Description:

The world's AI compute needs are accelerating and Arm is growing rapidly in the cloud. We work at the intersection of the two where the impact is global and open to all.

You will join a diverse, dedicated and inclusive group from all over the world based in Arm's modern offices in the vibrant city of Manchester. You will contribute directly to open source AI software projects, enabling them to use the full potential of Arm's line of Neoverse cores. Together with teams from across Arm, the open source community, and our partners you will be accelerating AI for the worlds most pervasive architecture.

We work on the full stack, from the high-level user interfaces, to graph compilers, libraries, and assembly; we encourage everyone in the team to get involved in all parts. Our team has a wide remit, but we are all aligned on a common goal. We trust and empower our team with the autonomy to identify and solve problems without silos and with the support of their team mates.

Key skills and experience:

• Strong software engineering and programming skills, preferably C++ and Python.
• Eager to dive into new and unfamiliar codebases to debug issues and look for performance opportunities.
• Familiarity with Linux development environments.
• A degree in a computational or numerate subject, or equivalent experience.

Additional skills and experience of interest:

• Experience with AI/ML frameworks and libraries: PyTorch; TensorFlow; ONNX; oneDNN etc.
• Experience in scientific and numerical computing, or data science.
• An understanding of parallel programming.
• Practical experience with assembly, particularly Arm SIMD (Neon, SVE, SME etc.).

Some of these key interpersonal strengths:

• Strong and clear communication skills.
• Experience delivering technical work involving multiple contributors.
• Contributing to open source projects.

In Return:

You will receive an attractive reward package with generous stock awards. We also know that people do their best work when they feel rested and psychologically safe, so we have a progressive leave policy including generous parental leave, days of care and 4 weeks of sabbatical every 4 years. We are flexible on working hours, and allow hybrid working. We strive to provide an open and accepting environment where you are encouraged to share your ideas and opinions; which enables collective innovation and creativity, and supports your professional and personal growth.