Skip to main content

Senior Software Engineer - CUDA Python

Senior Software Engineer - CUDA Python
Prodigy Game
remote
7 months 1 week ago

NVIDIA’s Invention and Vision

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.

Opportunity for Software Professionals

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.

  • 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

Qualifications

  • BS, MS or PhD degree in Computer Science, Electrical Engineering or related field (or equivalent experience)
  • 12+ 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

Stand Out

  • 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

The base salary range is 216,000 USD - 333,500 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. NVIDIA accepts applications on an ongoing basis.

Equal Opportunity Employer

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.

Expertise level

Work arrangement

Key skills

Similar Jobs in United States

Similar Jobs in