Senior Machine Learning Software Engineer

Overview

In this role you contribute to building and optimizing C++ and Python software for ML workloads on Arm-based platforms. You’ll work with ML software and hardware engineers to boost performance and efficiency, spanning areas like ML compilation, PyTorch integration, runtimes, and embedded software. You’ll take on technical design ownership and help tackle complex engineering challenges across the ML software stack. This opportunity combines software, ML, and hardware in a collaborative, growth-oriented environment.

Pay / Benefits

• comprehensive employee benefits

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, and embedded software
• Perform performance analysis and identify optimisation opportunities
• Contribute to technical design and own larger engineering challenges
• Support development of other engineers and foster collaboration

Key requirements

• Professional C++ development and experience with complex software systems
• Experience with ML frameworks or tooling such as PyTorch
• Python for software development, ML tooling, testing or automation
• Strong debugging and performance analysis skills
• Experience with collaborative software development using source control, code review, testing and CI

• Collaborative mindset and teamwork
• Problem-solving and debugging persistence
• Ownership and ability to drive initiatives
• C++ (professional development)
• Python
• PyTorch

…