A leading technology company is seeking software engineers for its math libraries teams focusing on AI and HPC kernel generation. The role involves developing algorithms for numerical dense linear algebra software on GPUs, providing leadership to library engineers, and collaborating with product management to meet performance requirements. Candidates should have a PhD or Master’s in a relevant field, along with strong CUDA and C++ skills. Competitive salaries and generous benefits are offered.
Qualifications
5+ years of experience in designing, developing, and optimizing production software.
Good knowledge of GPU or CPU hardware architecture.
Experience working with a globally-distributed team.
Responsibilities
Scope, design, and implement numerical dense linear algebra software on GPUs.
Provide leadership and feedback to engineers and mentor interns.
Collaborate with product management to define feature requirements.
Find opportunities to improve performance and reduce maintenance overhead.
Skills
CUDA
C++
Finite precision arithmetics
Numerical methods for linear algebra
Teamwork
Communication
Education
PhD or Master’s degree in Computer Science, Applied Math, or related field
Tools
CUTLASS
Python
Job description
A leading technology company is seeking software engineers for its math libraries teams focusing on AI and HPC kernel generation. The role involves developing algorithms for numerical dense linear algebra software on GPUs, providing leadership to library engineers, and collaborating with product management to meet performance requirements. Candidates should have a PhD or Master’s in a relevant field, along with strong CUDA and C++ skills. Competitive salaries and generous benefits are offered.