Jobs

Staff Software Engineer - Machine Learning


Job details
  • ARM
  • Newton
  • 1 week ago

Arm's Machine Learning (ML) Group is seeking a highly motivated and creative Software Engineer to join and lead a growing team of brilliant engineers located in Cambridge, UK specialising in ML frameworks and compilers.

This role presents an opportunity to contribute to sophisticated ML technology supporting Arm's ML Hardware. You will help to build the software that enables development of deep learning applications in many ground-breaking fields including self-driving cars, generative AI engines and ML-powered wearables.Job Description:

The ML Tooling team is looking for a software engineer with line management experience who can build a range of innovative compiler solutions for a variety of markets.You will apply your experience and insight to craft and optimise compilers for machine learning networks that target Arm’s CPUs, GPUs and NPUs.Responsibilities:

Contribute to deliver production-grade software and push the boundaries of Machine Learning compilationRecruit, develop, support and retain your engineers working as part of a larger teamBuild, extend and collaborate on innovative ML compilation software projects, such as TensorFlow, PyTorch, TOSA and the broader MLIR ecosystemWork with other groups in Arm to expand support for Arm architecture and ecosystemRequired skills and experience:

A passion for software development and qualityLine management of a small team (4-8 engineers)Proven experience with C++, understanding of Python is a plusExperience with the full software development lifecycle - planning, designing, developing, testing, delivering, and maintaining production-quality softwareExperience with or interest in compilersDesire to learn new skills and technologies and work in a highly motivated teamHigh degree of initiative and problem solving skillsAbility to own team's delivery and lead others on large or more sophisticated tasksExcellent communication skills."Nice to have" skills and experience:

Experience with contributing to open-source projects and working with a broader open-source communityKnowledge or curiosity about computer vision, machine learning, their applications and frameworksExperience with Linux and scripting languages, such as shell-scriptingIn Return:

On top of the already compelling

Life at Arm , we offer strong team culture, learning opportunities, regular career conversations, emphasis on diversity, equity and inclusion and a continuous improvement approach.

#LI-JB1Accommodations 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

. 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.Hybrid Working at Arm

Arm’s approach to hybrid working is designed to create a working environment that supports both high performance and personal wellbeing. We believe in bringing people together face to face to enable us to work at pace, whilst recognizing the value of flexibility. Within that framework, we empower groups/teams to determine their own hybrid working patterns, 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.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 race, color, religion, sex, sexual orientation, gender identity, national origin, disability, or status as a protected veteran.

Sign up for our newsletter

The latest news, articles, and resources, sent to your inbox weekly.

Similar Jobs

Staff Software Engineer

Job Overview:Our group writes the layer of the driver that implements the popular GPU compute APIs, such as OpenCL, which provide massively parallel execution of general purpose (C-like) code on Mali GPUs, often known as GPGPU (General-purpose computing on graphics processing units). Our many customers have performance-critical GPU compute use...

Cambridge

Staff Software Engineer

Job Overview:Our group writes the layer of the driver that implements the popular GPU compute APIs, such as OpenCL, which provide massively parallel execution of general purpose (C-like) code on Mali GPUs, often known as GPGPU (General-purpose computing on graphics processing units). Our many customers have performance-critical GPU compute use...

ARM Newmarket

Staff Software Engineer

Job Overview:Our group writes the layer of the driver that implements the popular GPU compute APIs, such as OpenCL, which provide massively parallel execution of general purpose (C-like) code on Mali GPUs, often known as GPGPU (General-purpose computing on graphics processing units). Our many customers have performance-critical GPU compute use...

ARM Cambridge

Staff Software Engineer

Job Overview:Our group writes the layer of the driver that implements the popular GPU compute APIs, such as OpenCL, which provide massively parallel execution of general purpose (C-like) code on Mali GPUs, often known as GPGPU (General-purpose computing on graphics processing units). Our many customers have performance-critical GPU compute use...

ARM Haverhill

Staff Software Engineer - Machine Learning (Some experience required)

Arms Machine Learning (ML) Group is seeking a highly motivated and creative Software Engineer to join and lead a growing team of brilliant engineers located in Cambridge, UK specialising in ML frameworks and compilers.You could be just the right applicant for this job Read all associated information and make sure...

ARM Farnham

Staff Software Engineer - Machine Learning

Arm's Machine Learning (ML) Group is seeking a highly motivated and creative Software Engineer to join and lead a growing team of brilliant engineers located in Cambridge, UK specialising in ML frameworks and compilers.This role presents an opportunity to contribute to sophisticated ML technology supporting Arm's ML Hardware. You will...

ARM Newton