Senior Software Engineer, CUDA Core Libraries

NVIDIA Ltd.
Santa Clara, United States of America
2 days ago

Role details

Contract type
Permanent contract
Employment type
Full-time (> 32 hours)
Working hours
Regular working hours
Languages
English
Experience level
Senior
Compensation
$ 288K

Job location

Santa Clara, United States of America

Tech stack

API
C++
Software Documentation
Profiling
Code Review
Nvidia CUDA
Computer Programming
Computer Engineering
Continuous Integration
Programming Tools
Python
Performance Tuning
Software Engineering
Openapi
Rust
Graphics Processing Unit (GPU)
PyTorch
Gpu Programming
Information Technology
Build Tools
Api Design
C++14

Job description

  • Design and implement idiomatic Python APIs and bindings for foundational CUDA capabilities and GPU algorithms.
  • Develop and integrate the native C/C++ components that support Python-facing functionality.
  • Define reliable and efficient interoperability boundaries between Python, C/C++, Rust, and other languages.
  • Develop high-performance interfaces that minimize Python and native-language integration overhead.
  • Own features throughout their lifecycle: design, implementation, testing, profiling, benchmarking, documentation, release, and long-term maintenance.
  • Improve the Python developer experience through typing, packaging, examples, diagnostics, continuous integration, and compatibility testing.
  • Collaborate with C/C++, Rust, compiler, and runtime engineers on shared architecture and API decisions.
  • Work directly with users to investigate correctness, usability, compatibility, and performance issues.

Requirements

We are hiring a Senior Software Engineer to advance the Python experience for CUDA Core Libraries. You will build Pythonic APIs, language bindings, algorithms, and runtime infrastructure on top of native C/C++ foundations. You will join the team building the foundational libraries, algorithms, and language/runtime infrastructure that make CUDA a speed-of-light experience for developers and AI coding agents alike., * BS, MS, or PhD in Computer Science, Computer Engineering, or a related field, or equivalent experience.

  • 8+ years of relevant software-development experience.
  • Strong production programming skills in both Python and C/C++; both are required for this role.
  • Experience building Python interfaces to native or systems-level software.
  • Understanding of systems software concepts, performance, concurrency, and API design.
  • Practical experience with parallel, heterogeneous, or GPU programming.
  • Experience developing production software or widely used libraries, including testing, profiling, benchmarking, packaging, and code review.
  • Ability to work independently, define project scope, and drive complex work to completion.
  • Clear written communication skills for API specifications, technical designs, and user documentation.
  • Comfort working in large codebases spanning Python, C/C++, build systems, packaging, and continuous-integration infrastructure.

Ways to stand out from the crowd:

  • Strong understanding of CPU/GPU architecture and performance optimization, with hands-on experience in GPU-accelerated stacks (CUDA C++/Python, PyTorch, JAX, Numba, CuPy, or similar).
  • Proficiency with modern C++ and GPU libraries such as Thrust, CUB, and libcudacxx.
  • Experience with compiler infrastructure and tooling, including LLVM, Clang, or MLIR.
  • Expertise in designing low-overhead interoperability between Python and native languages, including exposure to Rust in mixed-language stacks.
  • Demonstrated interest in developer tools, library design, and improving developer productivity.

Benefits & conditions

Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5.

About the company

NVIDIA's accelerated computing platform is foundational to modern HPC and AI. At the center of this platform are CUDA Core Libraries that enable developers to build fast, reliable, and scalable GPU-accelerated software.

Apply for this position