Machine Learning Engineer - PyTorch/C++ Development (NPU Architecture Team)
Actively Hiring
Full-time Posted 5 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.
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...