We are looking for experienced software engineers to join our team bringing delightful developer and user experiences to the Python compiler ecosystem. Our goal is to grow NVIDIA’s accelerated Python offerings to a make Python one of the first-class citizens for programming NVIDIA CUDA GPUs.

You will be a 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 AI, 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 enhance the functionality and performance of NVIDIA GPUs such that Python users can 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

• Address unique challenges in developing and deploying Python GPU solutions

• Engage with the Python community 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)

• 6+ years experience in compiler development and code optimization

• Strong Python programming skills with track record of driving formulation or adoption of community standards

• Fluent C, C++, and CUDA C++ programming skills

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

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

• Knowledge of Numba, NumPy, SciPy or a similar framework

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 and dynamic analysis techniques, and dynamic code generation

• Experience developing with the LLVM and MLIR 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

NVIDIA’s invention of the GPU revolutionized parallel computing. 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. Our GPUs are being used in many of the largest high-performance computing projects around the world, solving real world problems. Our products are used to build and parallelize the most meaningful scientific applications for weather modeling, climatology, fluid dynamics and defense. We support real science and scientists throughout the world.

With competitive salaries and a generous benefits package,NVIDIA is widely considered to be one of high technology’s most desirable employers. We have some of the most forward-thinking and experienced people in the world working for us. Our goal is to craft an environment where we can do our life’s best work. If you’re creative, autonomous, and highly motivated, we want to hear from you!

The base salary range is 184,000 USD - 356,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.