Expoint - all jobs in one place

Finding the best job has never been easier

Limitless High-tech career opportunities - Expoint

Nvidia Senior Math Libraries Engineers - Python APIs 
United States, Texas 
579809225

24.06.2024

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.

You will also be eligible for equity and .