Software Engineer in Machine Learning Tools

Overview

Job Overview:

Arm’s Machine Learning (ML) Group is seeking a highly motivated and creative Software Engineer to join a team of brilliant engineers who specialise in ML model optimisation tools.

This role presents an excellent opportunity to contribute to ground breaking Machine Learning technology. You will help build the software that enables development of deep learning applications in gaming, smart devices, home automation, self-driving cars and many other areas. Remember, wherever computing happens, Arm is there! You truly will have a chance to leave your mark on the future of AI, and what empowers it.

We are looking for candidates for this role at Staff or Senior Engineer grade and will agree with you the most appropriate grade during the hiring process based on your skills and experience.
Responsibilities

You will be responsible for work on exciting technology, lead implementation of new algorithms and develop advanced optimisations, allowing outstanding performance. The ML… tooling team contributes to open source and proprietary software tools and ML frameworks to help developers get the best performance out of their AI/ML applications on Arm hardware: CPUs, GPUs and neural network accelerators (NPUs).

You will collaborate with other teams in Arm and the open source community, such as Tensorflow and Pytorch, to deliver software enabling our partners to develop their next generation applications. You will play a leading role in achieving our goal to make life easier for Machine Learning developers.
Required Skills and Experience
• Strong knowledge of C++ or Python
• Experience developing for at least one Machine Learning framework: Tensorflow, Keras, PyTorch, ONNX, Jax etc
• Understanding of Machine Learning workflows.
• A passion for software development and quality.
• Ability to own team’s delivery and lead others on large or more sophisticated tasks.
• High degree of initiative and problem solving skills.
• Good interpersonal and communication skills.

More than anything else, you must be driven to learn new techniques and develop new technologies. Given the impact and reach of this field, you’ll have opportunities to share this knowledge with other specialists across the company.
“Nice To Have” Skills and Experience
• Strong knowledge of both C++ and Python
• Experience with Machine Learning implementation, including automated optimisation techniques.
• Familiarity with ML compiler infrastructures, such as MLIR or TVM.
In Return

You will get to influence the direction of our developer tools, contribute to the latest neural network acceleration technologies and influence how Machine Learning applications are optimised and run on billions of devices for years to come. You will work with an outstanding team of hard-working and knowledgeable people, advance your technical leadership and influencing skills and become a recognised authority in the field. Arm provides a rich Machine Learning roadmap in different domains and allows you to experience the demands from diverse use cases.

With offices around the world, Arm is an equal opportunity employer and value diversity at our company. We do not discriminate on the basis of race, religion, colour, national origin, sex, gender, gender expression, sexual orientation, age, marital status, veteran status, or disability status. By enabling a vibrant, inclusive, meritocratic, and open workplace, where all our people can grow and succeed, we encourage our people to share their unique contributions to Arm’s success in the global marketplace.

All Arm employees are provided with vital training to succeed in their respective roles. As well as a friendly and high-performance working environment, Arm offers a competitive reward package, private healthcare and wellness support, as well as other benefits such as a supplementary pension, generous annual leave, parental leave, and sabbatical. There’s even an on-site gym in Cambridge and social events organised within the company! We are offering a hybrid approach to home and office working to provide an adaptable experience for all employees. We expect some working time to be spent in office, to promote a strong collaborative environment with good team integration but are accommodating to different home working requirements.

If the above resonates with you, we would love to talk!

#LI-JB1

Company:

arm

Qualifications:

Language requirements:

Specific requirements:

Educational level:

Level of experience (years):

Senior (5+ years of experience)

Tagged as: Industry, Machine Learning, Neural Networks, NLP, OCR, United Kingdom