Senior Python GPU Libraries Engineer for HPC Equity

NVIDIA Corporation; Santa Clara, Northern (CA, KY); Hybrid; USD 184,000 - 357,000; Full time

Job summary

NVIDIA Corporation is seeking an experienced software professional to contribute to design and development of accelerated and distributed Python APIs for numerical computing. Python dominates AI, data science and HPC, with NumPy, SciPy, TensorFlow and PyTorch.

Join our team to develop GPU-accelerated Python libraries, optimize performance, and enable distributed workflows from supercomputers to the cloud. You will design Python APIs, work with partners, and ensure production quality code.

Qualifications

• Architect and implement accelerated Python numerical libraries.
• Analyze performance across CPU and GPU architectures.
• Prototype integrations into targeted frameworks.
• Write maintainable production-grade code.
• Contribute to runtime systems for multi-GPU computing.

Responsibilities

• Work with product management and partners to understand use cases and contribute to technical roadmaps of libraries.
• Architect, prioritize, and develop accelerated and distributed implementations of numerical algorithms.
• Design future-proof Python APIs for accelerated numerical/scientific computing libraries.
• Analyze and improve performance of developed APIs on CPU/GPU architectures, including end-to-end workflows.
• Prototype integrations of developed APIs into targeted frameworks.
• Write effective, maintainable, and well-tested production code.
• Contribute to development of runtime systems underpinning multi-GPU computing at NVIDIA.

Skills

Python

C++

CUDA

Numerical methods

Parallel programming

Performance analysis

Education

BS/MS/PhD in CS/Math/EE

Tools

CuPy

Jax.NumX

NumX

cuNumeric

NumS

Job description

NVIDIA Corporation is seeking an experienced software professional to contribute to design and development of accelerated and distributed Python APIs for numerical computing. Python dominates AI, data science and HPC, with NumPy, SciPy, TensorFlow and PyTorch.

Join our team to develop GPU-accelerated Python libraries, optimize performance, and enable distributed workflows from supercomputers to the cloud. You will design Python APIs, work with partners, and ensure production quality code.