Staff Software ML Engineer
Core
Develop and optimize C++ and Python software for Machine Learning workloads, focusing on ML compilation, runtimes, drivers, and performance analysis to improve efficiency on Arm compute platforms.
Role type
Staff Software ML Engineer (ML Infrastructure)
Builds
ML frameworks, compilers, runtimes, and drivers for Arm compute platforms
Domain
Semiconductor / Machine Learning Infrastructure
Deliverable
production ML models
Required skills
C++, Python, Machine Learning frameworks (PyTorch), performance analysis, debugging, complex software systems
Preferred skills
ML compilers, PyTorch internals, embedded software, hardware acceleration
Technologies
C++, Python, PyTorch, Arm compute platforms
Responsibilities
Develop and optimize software for ML workloads, collaborate with ML software and hardware engineers, contribute to technical design, take ownership of engineering challenges
Seniority
Staff, technical leadership & mentorship
