Jobs

(Immediate Start) Principal Engineer - AI FrameworksPerformance (Some experience required)


Job details
  • ARM
  • Manchester
  • 6 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

SC Cleared MLOps Engineer

Looking for an MLOps Engineer with active SC Clearance to join a Public Sector client of ours~Essential Skills- AI/ML Ops Engineer- Strong knowledge of AI/MLProgramming languages such as Python, and experience with AI/ML frameworks (e.g., TensorFlow, PyTorch)Cloud experience (Azure is preferable)Kubernetes CI/CD Data monitorisationDetails- 3-6 month contract Outside IR35 Fully...

London

Senior Customer Analyst

Senior Customer Analyst - Retail Brand - Bradford 2 days per weekSalary £45k + a great package and benefits!We are excited to partner to a leading retail brand based in Bradford as they look for a new Senior Customer Analyst to join their in-house team.As the Senior Customer Analyst, you'll...

Leeds

Machine Learning Engineer

Ready to leverage your mastery of LLMs to drive productivity? At Trilogy, we're opening doors to an exceptional tech career, welcoming those who've honed their AI skills to elevate their expertise in a dynamic environment. We're offering a rare chance where your primary focus will be to further expand your...

Preston

Machine Learning Engineer

Ready to leverage your mastery of LLMs to drive productivity? At Trilogy, we're opening doors to an exceptional tech career, welcoming those who've honed their AI skills to elevate their expertise in a dynamic environment. We're offering a rare chance where your primary focus will be to further expand your...

Northampton

Principle Data Scientist

Job Title: Principal Data ScientistRole Overview:We are seeking a Principal Data Scientist to lead AI/ML initiatives, drive analytics growth, and work closely with C-suite clients. You’ll define strategic challenges, develop solutions using AI, machine learning, and data science, and mentor teams to deliver high-impact projects.Key Responsibilities:Lead AI/ML projects and create...

Farringdon Without

Data Scientist / Software Develpoer

Data Scientist / Software DeveloperLocation: Hybrid (Various Locations)Company: Global Engineering Services ProviderJoin a leading engineering services company and help solve complex problems using data science, AI, and software development. As part of a dynamic team, you'll work on projects ranging from optimizing drug trials to satellite control systems.Responsibilities:Analyze data and...

Farringdon Without