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 drive future CPU architecture development and serve as deliverables to Central Engineering for final production.
Responsibilities: This position is part of a dedicated team within the CT-ML group focused on analyzing ML workloads and rapidly prototyping highly optimized CPU kernels to enhance model performance and accuracy.
Required Skills and Experience: - Strong interest and passion for implementing high-performance kernel code in dynamic environments.
- 4+ years of experience in implementing high-performance CPU kernels with vector and matrix extensions.
- Experience measuring and understanding performance metrics.
- Experience in creating efficient kernel development frameworks, including tools and testing methodologies.
- 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.
In Return: Arm offers an attractive relocation package and is committed to global talent acquisition. With offices worldwide, Arm fosters a diverse, inclusive, meritocratic, and open workplace where employees can grow and succeed. We encourage our people to share their contributions to Arm's success in the global marketplace.
Accommodations at Arm We support our candidates' needs during recruitment. If you require assistance or accommodations, please email . Your information will be kept confidential and used solely to provide appropriate support. Examples include breaks, document reading, or office accessibility. Please contact us to discuss your needs.
Hybrid Working at Arm Our hybrid work model aims to balance high performance with personal wellbeing. Teams can determine their own hybrid schedules based on work and needs. Details will be shared upon application, and we will work with you to find the best solution considering local regulations and circumstances.
Equal Opportunities at Arm Arm is an equal opportunity employer committed to a respectful environment. We do not discriminate based on race, color, religion, sex, sexual orientation, gender identity, national origin, disability, or veteran status.