Senior Software Engineer

Arm’s Machine Learning Group is seeking highly motivated and creative Software Engineers to join the Cambridge-based applied ML team!

From research, to proof-of-concept development, to deployment on ARM IPs, joining this team would be a phenomenal opportunity to contribute to the full life-cycle of machine learning projects and understand how state-of-the-art machine learning is used to tackle real word problems.

Working closely with experts in ML and software and hardware optimisation – a truly multi-discipline environment – you will have the chance to explore existing or build new machine learning techniques, while helping unpick the complex world of use-cases that are applied on high end mobile phones, TVs, and laptops.

About the role

The role would be part of a team passionate about understanding and following the latest trends in machine learning neural networks, collecting and analysing these use cases to drive the direction of ML workload support in products.

Your role is to work with major ML frameworks (PyTorch, Tensorflow, etc) to port and develop ML use cases, condition and quantise models to run more efficiently on Arm platforms, and to help ensure multiple Arm products are moulded to work well for machine learning.

As an in-depth technical role, you will need to understand the complex applications you analyse in detail and communicate them in their simplest form to help include them in product designs, where you will be able to influence both IP and system architecture.

About you

The ideal person is someone with a keen problem solving mind, passionate about state of the art machine learning, and its ability to find solutions to new and complex problems. As a dynamic field, being able to provide a clear view of the different options for solving each problem and the trade-offs between them will be the biggest challenge.

Essential skills and experience:

• A background in computer science, software engineering or other comparable skillset
• Experience training and debugging neural networks with TensorFlow or PyTorch using Python
• Familiarity with Linux and cloud services
• Have a strong attention to detail to ensure use cases you investigate are well understood and the critical areas needing improvement are understood
• Show strong communication skills – in this role you will regularly share your work with wider teams within Arm

Desirable skills:

• Experience of developing and maintaining CI/testing components to improve automation of model analysis
• Experience of the inner workings of Tensorflow or Tensorflow Lite
• Good knowledge of Python for working with ML frameworks
• Good knowledge of C++ for working with optimised ML libraries
• Previous experience of machine learning projects
• Experience with deployment optimizations on machine learning models