Job Overview:
High-performance ML workloads on Arm CPUs require the co-development of algorithms and highly optimized CPU kernels. In CT-ML (Central Technology, Machine Learning), rapid kernel prototyping is crucial for exploring algorithms and assessing trade-offs between model accuracy and performance. Successful prototypes are essential to drive future CPU architecture development and deliverables to Central Engineering for final production.
Responsibilities:
This position is part of a dedicated team within the CT-ML group focusing on analyzing ML workload and rapid prototyping of highly optimized CPU kernels to drive model performance and accuracies.
Required Skills and Experience:
* Strong interest and passion for implementing high-performance kernel code in a dynamic environment.
* 4+ years of experience in implementing high-performance CPU kernels with vector and matrix extensions.
* Experience measuring and understanding performance.
* Experience in creating an efficient kernel code development framework including tools and testing.
* Deep understanding of CPU architecture.
“Nice To Have” Skills and Experience:
* Knowledge of ML models and algorithms is a plus.
* Advanced degree or equivalent experience in Computer Architecture and Software are a plus.
In Return:
Arm is committed to global talent acquisition, offering an attractive relocation package. With offices around the world, Arm is a diverse organization of dedicated, creative, and highly talented engineers. By enabling a dynamic, inclusive, meritocratic, and open workplace, where all our people can grow and succeed, we encourage our people to share their unrivaled contributions to Arm's success in the global marketplace.
Accommodations at Arm:
At Arm, we want our people to Do Great Things. If you need support or an accommodation to Be Your Brilliant Self during the recruitment process, please email accommodations@arm.com. All accommodation requests will be treated with confidentiality, and information concerning these requests will only be disclosed as necessary to provide the accommodation.
Hybrid Working at Arm:
Arm’s approach to hybrid working is designed to create a working environment that supports both high performance and personal wellbeing. We believe in bringing people together face to face to enable us to work at pace while recognizing the value of flexibility. Within that framework, we empower groups/teams to determine their own hybrid working patterns, depending on the work and the team’s needs.
Equal Opportunities at Arm:
Arm is an equal opportunity employer, committed to providing an environment of mutual respect where equal opportunities are available to all applicants and colleagues. We are a diverse organization of dedicated and innovative individuals, and we don’t discriminate on the basis of race, color, religion, sex, sexual orientation, gender identity, national origin, disability, or status as a protected veteran.
#J-18808-Ljbffr