Senior Software Engineer - CUDA Python

Job Description

NVIDIA is looking for experienced software professionals to lead the development of delightful developer and user experiences in the Python ecosystem, focusing on NVIDIA's accelerated Python offerings for CUDA GPUs.

Requirements

• BS, MS or PhD degree in Computer Science, Electrical Engineering or related field
• 5+ years of relevant industry experience or equivalent academic experience after BS
• Strong Python programming and deployment skills with track record of driving formulation and/or adoption of Python community standards
• Fluent C/C++ and CUDA programming skills
• Background in AI, high performance computing or performance critical applications
• Experience in designing, developing, tuning, navigating, and/or 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
• 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