Job Information
Nvidia Senior Software Engineer - CUDA Python in Santa Clara, California
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 180,000 USD - 339,250 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions.
You will also be eligible for equity and benefits (https://www.nvidia.com/en-us/benefits/) . NVIDIA accepts applications on an ongoing basis.
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.
NVIDIA is a Learning Machine
NVIDIA pioneered accelerated computing to tackle challenges no one else can solve. Our work in AI and the metaverse is transforming the world's largest industries and profoundly impacting society.
Learn more about NVIDIA .