Staff Engineer - ML Libraries Development Engineer
Core
Optimizing core ML kernels and linear algebra functions for AI/ML acceleration on Qualcomm SoC CPUs using ARM architecture advancements.
Role type
Staff Engineer, ML Libraries Development
Builds
Core math libraries for ML/AI acceleration on Qualcomm SoC
Domain
Semiconductor / CPU Architecture / Machine Learning
Deliverable
production ML models
Required skills
ARM CPU architecture fundamentals, ARM Arch64 ISA, Vector processor kernel optimization, Linear algebra functions, Algorithm design, Performance evaluation and optimization, PyTorch/TensorFlow/Keras inference, Open-source library framework design
Preferred skills
Deep understanding of ARM ISA, Vector and matrix accelerator algorithms, Analytical and debugging skills, Linear algebra algorithm optimization, Performance evaluation on QEMU/Simulators/Emulators/Real Hardware
Technologies
ARM, SME, SVE, PyTorch, TensorFlow, Keras, QEMU, C, C++, Java, Python
Responsibilities
Optimize core ML kernels using ARM CPU architecture advancements (SME, SVE), Enhance performance of ML models on Qualcomm SoC CPU, Design algorithms for logic and critical thinking, Evaluate and optimize applications for ARM architecture, Infer ML models written in PyTorch/TensorFlow/Keras, Understand Open-Source Library framework design
Seniority
Staff, 8-12 years experience
