Senior Software Engineer - Machine Learning Architecture
Design and optimize machine learning kernels for ARM architecture, collaborating with multiple teams. Requires strong programming skills, linear algebra knowledge, and experience with PyTorch/TensorFlow.
Key Highlights
Technical Skills Required
Benefits & Perks
Job Description
- This role is on-site 5 days per week in San Diego, CA and relocation packages are available for the right candidates. Candidate would participate in software architecture, Optimizing the core ML kernels using the latest advancements in ARM architecture features like SME, SVE that enhance the ML models on the CPU of the SoC.
- Bachelor's degree in Engineering, Information Systems, Computer Science, or related field and 2+ years of Software Engineering or related work experience.
Master's degree in Engineering, Information Systems, Computer Science, or related field and 1+ year of Software Engineering or related work experience.
OR
PhD in Engineering, Information Systems, Computer Science, or related field.
- 2+ years of academic or work experience with Programming Language such as C, C++, Java, Python, etc.
- Master's Degree in Engineering, Information Systems, Computer Science or related field.
- Understanding of ARM CPU architecture fundamentals and ARM ISA
- Understanding of the basic linear algebra functions used in AI/ML
- Performance Evaluation and Optimization of the applications for ARM architecture
- Inferencing of the ML models written in Pytorch/TensorFlow
- Understanding of the typical Open Source Library framework design
- Skills to collaborate with multiple-teams including HW design to define the software architecture
- Good Programming skills and understanding of the ARM ISA
- understanding of the algorithms suitable for Vector and Matrix accelerators
- Strong Analytical and debugging skills
Compensation is determined by several factors which may include skillset, experience level, and geographic location.
The expected range for this role is $110,000 to $135,000 per year. Please note this range is an estimate and actual pay may vary based on qualifications and experience.
Similar Jobs
Explore other opportunities that match your interests