Jobs

Senior/Staff Software Engineer - Machine Learning Frameworks(Some experience required)


Job details
  • ARM
  • Manchester
  • 3 days ago

Job Overview: Are you a skilled and motivated engineerwith a strong background in optimised system design such asdistributed systems and high-performance concurrency? We arebuilding the future of computing on Arm. We want to make Arm-basedhardware the natural choice for ML in the datacenter. To that end,we truly believe that major machine learning frameworks used to runAI must be highly performant. Our team is a diverse, dedicated, andinclusive group from all over the world based in Arms stylishoffices at the heart of Manchester. We work on all aspects ofresearching, developing, and delivering highly optimised MLframeworks into the Arm ecosystem across many ML models. This rolewill directly contribute to key open source ML frameworks such asTensorFlow and PyTorch. In addition, Arm is the owner and advocateof the underlying technologies, such as Compute Library, that actas basic building blocks to form the high-quality and performantsoftware. In collaboration with colleagues from Manchester andCambridge, you will work on delivering optimised software forserver-class hardware and integrate it with ML software frameworksand libraries for deployment on our partners hardware. We work withexciting technology, help to implement new algorithms, and optimisefor the latest Arm server hardware. Our work has a high impact inthe ML ecosystem, with the possibility to engage with partners andthe community. Responsibilities: Your role as a Senior SoftwareEngineer will require you to: ● Optimise ML software to utilise thefull potential of Arms line of Neoverse cores for datacentre andcloud use cases at high core counts. ● Work with development teamsbased from compilers and libraries to extend the capabilities ofCompute Library to meet the unique demands of ML workloads onservers. ● Improve Compute Library API to support flawlessintegration with ML frameworks. ● Extend ML frameworks to simplifyintegration with Compute Library. Required Skills and Experience: ●Software development, with very good programming skills, preferablyC++ and Python. ● Excellent understanding of parallel programmingprimitives and constructs. ● Ability to quickly investigate anddebug large software frameworks. ● Optimising code for performance.● A real passion for software development. ● Very goodinterpersonal, collaboration, and communication skills. ● Curiosityto make a positive impact, both in our team and in the wider Armecosystem. ● A degree, or higher, in a computational or numeratesubject, or experience in a related field. Note: Training may beprovided if you have only a subset of these key skills! “Nice ToHave” Skills and Experience: The following is a selection of skillsused across our projects. You do not need to have experience withany of these to apply or succeed in your application. ● Experiencewith one or more of: NEON, SVE, SVE2, and Arm assembler. ●Experience with TensorFlow, PyTorch, ONNXRuntime, andor oneDNN. ●Open source community citizenship including code commits andreviews. ● Basic Linux administration, particularly installationand maintenance. In Return: You will be joining an outstandingcompany! We strive to provide an open and accepting environmentwhere you are encouraged to share your ideas and opinions; whichenables collective innovation and creativity and supports yourprofessional and personal growth. In addition, we enjoy 25 days ofannual leave as well as progressive parental leave, support forflexible and hybrid working, and many other benefits that Armoffers. #LI-JB1 Accommodations at Arm At Arm, we want our people toDo Great Things. If you need support or an accommodation to Be YourBrilliant Self during the recruitment process, please . To note, by sending us the requestedinformation, you consent to its use by Arm to arrange forappropriate accommodations. All accommodation requests will betreated with confidentiality, and information concerning theserequests will only be disclosed as necessary to provide theaccommodation. Although this is not an exhaustive list, examples ofsupport include breaks between interviews, having documents readaloud, or office accessibility. Please email us about anything wecan do to accommodate you during the recruitment process. HybridWorking at Arm Arm’s approach to hybrid working is designed tocreate a working environment that supports both high performanceand personal wellbeing. We believe in bringing people together faceto face to enable us to work at pace, whilst recognizing the valueof flexibility. Within that framework, we empower groupsteams todetermine their own hybrid working patterns, depending on the workand the team’s needs. Details of what this means for each role willbe shared upon application. In some cases, the flexibility we canoffer is limited by local legal, regulatory, tax, or otherconsiderations, and where this is the case, we will collaboratewith you to find the best solution. Please talk to us to find outmore about what this could look like for you. Equal Opportunitiesat Arm Arm is an equal opportunity employer, committed to providingan environment of mutual respect where equal opportunities areavailable to all applicants and colleagues. We are a diverseorganization of dedicated and innovative individuals and don’tdiscriminate on the basis of race, color, religion, sex, sexualorientation, gender identity, national origin, disability, orstatus as a protected veteran.

Sign up for our newsletter

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

Similar Jobs

Senior/Staff Software Engineer - Machine Learning Frameworks(Some experience required)

Job Overview: Are you a skilled and motivated engineerwith a strong background in optimised system design such asdistributed systems and high-performance concurrency? We arebuilding the future of computing on Arm. We want to make Arm-basedhardware the natural choice for ML in the datacenter. To that end,we truly believe that major...

ARM Manchester

Senior/Staff Software Engineer - Machine Learning Frameworks (Some experience required)

Job Overview:Any additional information you require for this job can be found in the below text Make sure to read thoroughly, then apply.Are you a skilled and motivated engineer with a strong background in optimised system design such as distributed systems and high-performance concurrency?We are building the future of computing...

ARM Manchester

Staff/Senior Software Engineer - Machine Learning Compilers (Some experience required)

Arms Machine Learning (ML) Group is seeking a highly motivated and creative Software Engineer to join a team of brilliant engineers located in Cambridge, UK who specialise in ML compilers.This role presents an opportunity to contribute to advance ML compilation technology. You will help to build the software that enables...

ARM Cambridge

Staff/Senior Software Engineer - Machine Learning Compilers

Job DescriptionArm's Machine Learning (ML) Group is seeking a highly motivated and creative Software Engineer to join a team of brilliant engineers located in Cambridge, UK who specialise in ML compilers. This role presents an opportunity to contribute to advance ML compilation technology. You will help to build the software...

ARM Cambridge

Staff Software Engineer

Staff Software Engineer | Remote First (UK/EU) | Kotlin, Kafka, AWSStaff/Senior Software Engineer£90,000-£125,000 paSeed funded startup with product market fitRemote in the UK or EU⚙️Tech Stack: Kotlin (JVM), Event Driven, Kafka, AWSAre you ready to take on complex, real-time, large-scale challenges and shape the future of an innovative AI powered...

Burns Sheehan

Junior Software Engineer - Up to £45,000 (HYBRID) - LONDON

Junior Software Engineer - Up to £45,000 (HYBRID) - IN LONDON OFFICE Lorien's client, an upscaling firm operating in the world's largest financial market, is rolling out their graduate Software Developer programme for 2025. This is a fantastic time to join their team being responsible for the development of high-performance...

Lorien Resourcing Limited London