Staff Software Engineer – ML (Tooling)
Core
Design and develop software frameworks and tooling to validate ML workloads and measure performance across Arm's heterogeneous platforms (CPUs, GPUs, NPUs).
Role type
Staff Software Engineer (ML Tooling & Validation)
Builds
Automation pipelines, benchmarking frameworks, and visualization tools for ML system validation.
Domain
Semiconductor / Machine Learning Infrastructure
Deliverable
production ML models
Required skills
Python (OOP, uv, ruff), Linux/Mac development, Docker, CI/CD (Jenkins, GitLab, GitHub), System automation
Preferred skills
OS fundamentals, Pre-silicon/hardware bring-up, FPGA/Embedded experience, ML frameworks/compilers, Data visualization
Technologies
Python, Docker, Jenkins, GitLab, GitHub, Arm platforms, Linux, Mac
Responsibilities
Develop frameworks for running ML workloads and benchmarks; Automate execution, data collection, and visualization pipelines; Collaborate with ML software and systems teams to define requirements; Explore ML performance on future Arm architectures.
Seniority
Staff, hands-on IC with strategic influence
