Machine Learning Engineer - PyTorch/C++ Development (NPU Architecture Team)
Actively Hiring
Full-time Posted 7 months ago
Responsibilities
- check_circle Develop and enhance ML compilers using PyTorch and C++ tailored for Qualcomm’s Neural Processing Unit (NPU) architecture.
- check_circle Design and implement advanced quantization techniques to improve model efficiency and accuracy.
- check_circle Optimize ML workloads for deployment across diverse devices, ensuring top-tier performance and energy efficiency.
- check_circle Work directly with strategic customers to address specific challenges and deliver tailored ML solutions.
- check_circle Collaborate with architecture and software engineering teams to integrate cutting-edge ML technologies into Qualcomm platforms.
- check_circle Research, prototype, and implement innovative solutions in ML compiler design and system-level optimizations.
- check_circle Evaluate and debug ML performance to identify and resolve bottlenecks in complex workflows.
- check_circle Create thorough technical documentation to share knowledge and advancements with the team. Qualifications:
- check_circle Qualifications:
- check_circle Bachelor’s or advanced degree in Computer Science, Electrical Engineering, Machine Learning, or a related discipline.
- check_circle Strong expertise in PyTorch and C++ programming.
- check_circle Experience with ML workload analysis, compiler development, and quantization techniques.
- check_circle Familiarity with deep learning frameworks such as TensorFlow or ONNX is a plus.
- check_circle Proven track record of solving complex performance and efficiency challenges in hardware-aware ML solutions.
- check_circle Ability to work collaboratively with strategic customers and deliver impactful results.
- check_circle Excellent analytical skills and ability to thrive in a high-performance team environment.
Basic qualifications
- Bachelor's degree in Science, Engineering, or related field and 2+ years of ASIC design, verification, validation, integration, or related work experience.
- References to a particular number of years experience are for indicative purposes only. Applications from candidates with equivalent experience will be considered, provided that the candidate can demonstrate an ability to fulfill the principal duties of the role and possesses the required competencies.
Preferred qualifications
- Experience with ML model deployment on hardware accelerators such as GPUs, TPUs, or NPUs.
- Understanding of system-level architecture and low-level programming.
- Contributions to ML research or publications in relevant fields are an advantage.
Tags & Focus Areas
Machine Learning Pytorch Ai
About Qualcomm
Company: Qualcomm Canada ULC Job Area: Engineering Group, Engineering Group > Machine Learning Researcher General Summary: Qualcomm AI Research is looking for world-class researchers in deep learning, to join a high-caliber team of engineers inventing machine learning technology to provide best-in-class solutions while running with the most efficient use of power, memory, and computation. Minimum Qualifications: - Master's degree in Computer Engineering, Computer Science, Electrical Engineeri...