NVIDIA Job Opportunity: Specialist Software Engineer
NVIDIA is seeking a self-motivated and specialist software engineer to design and develop Python APIs for math libraries. The focus is on creating accessible math libraries for all developers within the Python ecosystem. The role involves crafting and developing Python APIs, prototyping integrations, performance tuning, optimization, and benchmarking of numerical software, providing technical leadership and mentorship, and collaborating with internal and external partners.
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
What We Need To See
- PhD or MSc degree in Computer Science, Applied Math, or related science or engineering field is preferred or equivalent experience
- 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
Join NVIDIA
NVIDIA is widely considered to be one of the technology world’s most desirable employers and is committed to fostering a diverse work environment. The base salary range is 176,000 USD - 333,500 USD, with eligibility for equity and benefits. NVIDIA accepts applications on an ongoing basis and is proud to be an equal opportunity employer. If you're creative and have initiative, this opportunity is for you!