Skip to main content

Staff Software Engineer - Machine Learning

Job ID 2024-11720 Date posted 24/10/2024 Location Cambridge, United Kingdom Category Machine Learning
Apply

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 compilation
  • Recruit, develop, support and retain your engineers working as part of a larger team
  • Build, extend and collaborate on innovative ML compilation software projects, such as TensorFlow, PyTorch, TOSA and the broader MLIR ecosystem
  • Work with other groups in Arm to expand support for Arm architecture and ecosystem

Required skills and experience:

  • A passion for software development and quality
  • Line management of a small team (4-8 engineers)
  • Proven experience with C++, understanding of Python is a plus
  • Experience with the full software development lifecycle - planning, designing, developing, testing, delivering, and maintaining production-quality software
  • Experience with or interest in compilers
  • Desire to learn new skills and technologies and work in a highly motivated team
  • High degree of initiative and problem solving skills
  • Ability to own team's delivery and lead others on large or more sophisticated tasks
  • Excellent communication skills.

"Nice to have" skills and experience:

  • Experience with contributing to open-source projects and working with a broader open-source community
  • Knowledge or curiosity about computer vision, machine learning, their applications and frameworks
  • Experience with Linux and scripting languages, such as shell-scripting

In 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-JB1

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.

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.

Life at Arm

  • Culture at Arm
    Culture at Arm

    Make an Impact

    We, not I. This belief is at the heart of Arm's company culture and it underscores the culture of collaboration alongside individual accountability in a supportive environment working together for the success of Arm. Across our entire ecosystem, we know that when you're able to be your most brilliant self, you can do great things.
    Read more

  • Diversity, Equity and Inclusion
    Diversity, Equity & Inclusion

    This is Collective Progress

    At Arm, we're committed to inspiring revolutionary ideas in a diverse, equitable, and inclusive environment. Be your most brilliant self, and empower others, via various avenues for active participation – Employee Resource Groups (ERGs), Employee Communities, DEI working groups, and DEI Council.
    Read more

  • Benefits at Arm
    Benefits at Arm

    Benefits Designed for You

    When our employees thrive, so does Arm. Because our teams are so remarkable, we offer remarkable benefits designed to nurture the professional and personal growth of the brilliant people building the future of computing.
    Read more

Jobs for You

  • Data Analytics Intern Develop dashboards, maintain data sets, and support data projects within the People Analytics team. Budapest, Hungary IT
  • Senior Applications Engineer Deliver training to tier-1 companies in the industry and become the technical expert for our customers by solving sophisticated issues. Hsinchu, Taiwan Applications Engineering
  • Staff Automation Engineer Develop and implement innovative IT automation solutions to enhance infrastructure efficiency, scalability, and security in a collaborative environment. Bengaluru, India IT

No previously viewed jobs

No jobs have been saved

Get Job Alerts

Can’t find the job you’re seeking? Register to be notified as soon as new jobs become available. Enter your email, select your preferred job category and/or location, then click Add to set your preferences and Subscribe to create your job alert.

Interested InSelect a job category from the list of options. Search for a location and select one from the list of suggestions. Finally, click “Add” to create your job alert.

  • Machine Learning, Cambridge, England, United KingdomRemove

By submitting your information, you acknowledge that you have read our privacy policy, and consent to receive email communication from Arm.

Join our Talent Community, Unlock Opportunities

Subscribe to receive Arm communications directly to your inbox. Stay connected to be the first to hear about updates from our community and exciting roles that align with your skills.

Join Now