Nvidia Senior Math Libraries Engineers - Python APIs 

United States, Texas 

579809225

NVIDIA is looking for a self-motivated and specialist software engineer for the design and development of Python APIs for math libraries.

What you will be doing:

• crafting and developing Python APIs for math libraries (e.g., linear algebra),

• prototype integrations of developed APIs into targeted frameworks,

• performance tuning, optimization, and benchmarking of numerical software on various CPU and GPU architectures,

• providing technical leadership and mentorship to library engineers working with you on projects,

• working closely with product management and other internal and external partners to understand feature and performance requirements and contribute to the technical roadmaps of libraries,

• finding opportunities to improve software performance and user experience.

see:

• PhD or MSc degree in Computer Science, Applied Math, or related science or engineering field is preferred or equivalent experience. Comparable relevant work experience will also be considered.

• Excellent Python and C++ programming and software design skills, including functional and performance tests design.

• Experience developing Python APIs and Pythonic style

• Experience using one or more of Numpy, CuPy, SciPy, JAX, PyTorch, TensorFlow.

• 6+ years experience developing, debugging, and optimizing high-performance applications on parallel computing platforms.

• Strong understanding of fundamental numerical methods and computations in science, engineering, or deep learning.

• Proven experience in leading and driving software development projects in an agile environment.

• Strong collaboration, communication, and documentation habits.

Ways to stand out from the crowd:

• Parallel programming experience with GPUs (CUDA or OpenCL), multi-threading, or MPI.

• Good knowledge of CPU and/or GPU hardware architecture.