Senior Math Libraries Engineer - Python API

NVIDIA Job Opportunity: Python Software Engineer

NVIDIA is seeking a self-motivated and specialist software engineer to design and develop Python APIs for math libraries. The company has been at the forefront of providing GPU accelerated Deep Learning frameworks, and aims to make these frameworks accessible to developers in the Python ecosystem.

Responsibilities

• Crafting and developing Python APIs for math libraries (e.g., linear algebra)
• Prototype integrations of developed APIs into targeted frameworks
• Performance tuning, optimization, and benchmarking of numerical software on various CPU and GPU architectures
• Providing technical leadership and mentorship to library engineers working on projects
• Working closely with product management and internal/external partners to understand feature and performance requirements
• Finding opportunities to improve software performance and user experience

Requirements

• PhD or MSc degree in Computer Science, Applied Math, or related field preferred
• Excellent Python and C++ programming and software design skills
• Experience developing Python APIs and Pythonic style
• 6+ years experience in developing high-performance applications on parallel computing platforms
• Strong collaboration, communication, and documentation habits

Preferred Qualifications

• Parallel programming experience with GPUs (CUDA or OpenCL), multi-threading, or MPI
• Good knowledge of CPU and/or GPU hardware architecture

About NVIDIA

NVIDIA is widely considered to be one of the technology world’s most desirable employers. The company fosters a diverse work environment and is an equal opportunity employer. Salary range is 176,000 USD - 333,500 USD, and equity and benefits are provided. NVIDIA accepts applications on an ongoing basis.