Senior Software Engineer - CUDA Python

Job Description

NVIDIA is seeking experienced software professionals to lead and extend their work on accelerating Python offerings for NVIDIA CUDA GPUs.

Requirements

• BS, MS or PhD degree in Computer Science, Electrical Engineering or related field
• 6+ years of relevant industry experience or equivalent academic experience after BS
• Strong Python programming and deployment skills
• Fluent C/C++ and CUDA programming skills
• Experience in AI, high performance computing or performance critical applications
• Experience in developing/maintaining Python projects and engaging with Python users on UX or performance improvements
• Experience in designing, developing, tuning, and maintaining a large, complex, multi-language software stack
• Good written communication, collaboration, and presentation skills
• Experience in distributed programming in C/C++/Python using MPI, Dask, Legate, or other distributed programming models/frameworks
• Knowledge of generating Python bindings for mid- to large-size C/C++ codebases

Responsibilities

• Architect, prioritize, and develop new features in CUDA Python
• Analyze, identify, and improve the UX and performance of CUDA software in Python
• Write effective, maintainable, and well-tested code for production use
• Bridge the language gap between existing CUDA C/C++ solutions and Python
• Understand and address unique challenges in developing and deploying Python GPU solutions
• Identify key open source players in the Python/PyData ecosystem, and engage with them to develop and drive necessary protocols and standards for the NVIDIA CUDA platform
• Evangelize CUDA programming in Python to encourage and empower adoption of the NVIDIA CUDA platform

Benefits

• Multiple relocation packages
• Two weeklong shutdowns (mid-summer and year-end) in the US (in addition to PTO)
• 8-week parental leave
• 9 Employee Resource Groups
• Annual bonus offering
• Flexible work arrangements
• Up to 6% 401K matching