Senior Software Engineer - CUDA Python

NVIDIA’s invention of the GPU in 1999 sparked the growth of the PC gaming market, redefined modern computer graphics, and revolutionized parallel computing. More recently, GPU deep learning ignited modern AI — the next era of computing — with the GPU acting as the brain of computers, robots, and self-driving cars that can perceive and understand the world. We're looking to grow our company, and form teams with the most inquisitive people in the world. Join us at the forefront of technological advancement.

We are looking for experienced software professionals to lead/extend our work on bringing delightful developer and user experience to the Python ecosystem. Our goal is to grow NVIDIA’s accelerated Python offerings to a mature product and make Python one of the first-class citizens for programming NVIDIA CUDA GPUs. You will be a crucial member of a team that is working to bring together the power of GPU acceleration and the expressibility and programmability of Python, by developing foundational software that supports many key products spanning the gamut of high performance computing, scientific computing, data analytics, deep learning, and professional graphics running on hardware ranging from gamer laptops to supercomputers to the cloud.

What You'll Be Doing:

As a member of our team, you will use your design abilities, coding expertise, creativity, and community engagement to develop and enhance the functionality and performance of NVIDIA GPUs such that the current and future generations of Python users can enjoy the programmability and take full advantage of the NVIDIA CUDA platform, including both NVIDIA hardware and software. Specifically, you will be working to:

• 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

What We Need To See:

• BS, MS or PhD degree in Computer Science, Electrical Engineering or related field (or equivalent experience)

• 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

• Track record of developing/maintaining Python projects, and/or engaging with Python users on UX or performance improvements

• Experience in designing, developing, tuning, navigating, and/or maintaining a large, complex, multi-language software stack (between C/C++/CUDA and Python)

• Good written communication, collaboration, and presentation skills with ability of operating across team boundaries

• 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

Ways To Stand Out From The Crowd:

• Deep understanding in the CUDA programming model and language features

• Familiarity with Python ecosystem, language idioms, and pioneering solutions

• Dexterity with compilers, static/dynamic analysis techniques, and/or dynamic code generation/transpilation/compilation

• Experience in using or developing the LLVM/Clang compiler infrastructure

• Experience in memory management of a multi-language project or development of domain specific libraries/languages for AI, Data Analytics or Scientific Computing

With competitive salaries and a generous benefits package, we are widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us and, due to unprecedented growth, our exclusive engineering teams are rapidly growing. If you're a creative and autonomous engineer with a real passion for technology, we want to hear from you!

The base salary range is 184,000 USD - 425,500 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.