Senior Software Engineer - NVIDIA Warp

at Nvidia
USD 184,000-356,500 per year
SENIOR
✅ On-site

Tech Stack

AI API CI/CD @ 4 CUDA @ 4 Communication @ 4 Data Pipelines @ 6 Debugging @ 4 GPU @ 4 Profiling @ 4 Python @ 4 Software Development @ 7

Details

NVIDIA Warp brings Python productivity to high-performance simulation, scientific computing, and differentiable programming on CPUs and NVIDIA GPUs. The team develops Warp for computational physics, AI, and optimization workflows and is seeking a hands-on senior engineer to expand Warp's adoption across science and engineering.

The goal is to make Warp a foundation for products and tools inside and outside NVIDIA by shaping the future of accelerated computing while making it easier to use.

Responsibilities

  • Own and improve the end-to-end technical path for adopting Warp in computational applications, from native C++ and CUDA changes through production integration, validation, and long-term compatibility.
  • Turn new CUDA features and programming models into supported Warp functionality that improves performance and portability across successive generations of NVIDIA accelerated-computing platforms and opens new application domains.
  • Partner with internal engineering teams, including those behind ALCHEMI, BioNeMo, PhysicsNeMo, and Kit-CAE, to integrate Warp into their software, resolve technical barriers, and support production adoption.
  • Select and integrate CUDA-X libraries and other computing technologies when they address recurring technical gaps, exposing them through consistent Python and native APIs.
  • Design and implement Warp improvements that expand what researchers and developers can accomplish, including new APIs, programming abstractions, and reusable building blocks for simulation, optimization, and AI workloads.

Requirements

  • A BS, MS, or PhD in Computational Science and Engineering, Computational Physics, Electrical Engineering, Computer Engineering, Mechanical Engineering, Aerospace Engineering, Computer Science with a focus on scientific computing, or a related computational engineering field, or equivalent experience.
  • 8+ years of relevant software development experience in industry, academic research, or a combination of both.
  • Significant professional or academic research experience building, debugging, profiling, and optimizing performance-critical CUDA C++ software for NVIDIA GPUs, including diagnosing performance and correctness issues involving parallel execution, GPU architecture, and memory behavior.
  • Substantial work building, optimizing, and supporting computational software used in research or production, involving numerical methods, model validation, or scientific data pipelines.

Preferred Qualifications

  • Ownership of framework integrations from prototype through production across multiple software projects.
  • Hands-on experience designing and optimizing software for systems that combine CPUs, GPUs, memory, and high-speed interconnects, including data placement, communication, and architecture-aware performance tuning.
  • Experience defining and evolving Python GPU programming systems across user-facing programming models, compilers, runtimes, and kernel libraries.
  • Experience shipping and maintaining production-quality open-source software, including testing, CI/CD, packaging, compatibility, and release practices within a large distributed team.
  • Hands-on experience developing, extending, or integrating commercial or open-source simulation software for multiphysics, CFD, semiconductor process or device simulation (TCAD), molecular dynamics, or related computational workloads. Examples include Ansys Fluent, Siemens Simcenter STAR-CCM+, COMSOL Multiphysics, OpenFOAM, LAMMPS, and GROMACS.

Compensation and Benefits

  • Base salary range of USD 184,000–287,500 for Level 4.
  • Base salary range of USD 224,000–356,500 for Level 5.
  • Eligibility for equity and benefits.
  • NVIDIA offers a comprehensive benefits package.

Applications will be accepted at least until August 15, 2026. This posting is for an existing vacancy. NVIDIA uses AI tools in its recruiting processes and is an equal opportunity employer.

More jobs at Nvidia

Similar jobs