Senior ML Software Engineer (C++ & PyTorch)

Arm Limited; Town Of Cambridge; Hybrid; AUD 120,000 - 160,000; Full time

Job summary

Arm is seeking experienced software engineers with strong C++ and Python skills to advance ML workloads across Arm compute platforms. You'll work with ML software and hardware engineers to optimize how models execute and where performance can be improved.

Opportunities span ML frameworks, compilers, runtimes, drivers, and embedded software, with seniority levels and technical leadership aligned to experience. Arm supports hybrid working and continuous learning.

Qualifications

• Professional C++ development and complex software systems.
• Machine Learning frameworks or tooling such as PyTorch.

Responsibilities

• Develop and optimise C++ and Python software for ML workloads.
• Collaborate with ML software and hardware engineers to improve model execution performance.
• Contribute to ML compilation, PyTorch integration, runtimes and drivers, or embedded software.
• Take ownership of larger engineering challenges and mentor others.

Skills

C++ development

PyTorch

Python

Job description

Arm is seeking experienced software engineers with strong C++ and Python skills to advance ML workloads across Arm compute platforms. You'll work with ML software and hardware engineers to optimize how models execute and where performance can be improved.

Opportunities span ML frameworks, compilers, runtimes, drivers, and embedded software, with seniority levels and technical leadership aligned to experience. Arm supports hybrid working and continuous learning.