CUDA Python Technical Product Manager

US, CA, Santa Clara

NVIDIA

NVIDIA erfindet den Grafikprozessor und fördert Fortschritte in den Bereichen KI, HPC, Gaming, kreatives Design, autonome Fahrzeuge und Robotik.

View all jobs at NVIDIA

Apply now Apply later

Are you passionate about using Python to bring to bear high-performance computing? Do you want to help Python developers unlock the full potential of GPUs on the CUDA platform? As the demand for GPU-accelerated applications grows across industries like AI, scientific computing, and data analysis, Python developers increasingly seek a more native, detailed experience with CUDA. We are looking for a Technical Product Manager who can drive the vision, strategy, and product roadmap to bring Python to the CUDA platform as a native language.

As a Technical Product Manager for CUDA Python, you will work closely with internal engineering teams, the open-source community, and external developers to define and prioritize features, ensure compatibility with the Python ecosystem, and streamline GPU programming for millions of Python developers worldwide. You'll help craft the future of high-performance Python by providing a detailed experience for Python developers on CUDA while advocating for their needs within NVIDIA.

What you’ll be doing:

  • Define the product vision, strategy, and roadmaps for CUDA Python

  • Collaborate with internal CUDA engineers, Python core developers, and key team members to deliver a seamless Python-to-CUDA experience.

  • Engage with the open-source community, understand their needs, and translate those into product features that improve the developer experience.

  • Ensure the integration of CUDA Python aligns with other key NVIDIA technologies

  • Identify and prioritize use cases for CUDA Python across industries like machine learning, scientific computing, and deep learning.

  • Work with developer relations and marketing teams to craft a go-to-market strategy, positioning CUDA Python as a critical tool for developers, scientists, and researchers.

  • Analyze market trends and customer feedback to make data-driven decisions on product features and improvements.

What we need to see:

  • BS, or MS in Computer Science, Computer Engineering, or a similar field, or equivalent experience.

  • 6+ years of experience in product management, technical project management, or similar roles in the technology industry.

  • Solid understanding of GPU computing, CUDA, and Python programming.

  • Experience working with developers, open-source communities, or technical users to gather feedback and drive product improvements.

  • Excellent communication and interpersonal skills with a strong ability to influence cross-functional teams.

Ways to stand out from the crowd:

  • Experience working on new technologies, particularly with Python and GPU computing, and bringing them to market.

  • In-depth knowledge of CUDA, Python libraries for numerical computation (such as NumPy, SciPy, or cuPy), and GPU-accelerated machine learning frameworks like TensorFlow or PyTorch.

  • Familiarity with compiler technologies, JIT compilation, and performance optimization for high-performance computing.

  • Passion for driving community engagement and improving developer tools.

If you thrive with solving problems and are passionate about Python, we'd love to hear from you! Join us to craft the future of GPU programming. With highly competitive salaries, a comprehensive benefits package, and an excellent engineering work culture, NVIDIA is widely considered to be one of the industry's most desirable employers.

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

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.

Apply now Apply later
Job stats:  0  0  0

Tags: Computer Science CUDA Data analysis Deep Learning Engineering GPU Machine Learning NumPy Open Source Python PyTorch SciPy TensorFlow

Perks/benefits: Career development Competitive pay Equity / stock options

Region: North America
Country: United States

More jobs like this