Requirements:
- BS/MS (or equivalent experience) in CS/CE/EE/Physics/Applied Math or related field.
- Strong programming skills in C/C++ and Python on Linux.
- Hands-on experience with CUDA programming and GPU performance optimization concepts.
- Experience profiling and debugging performance using tools such as NVIDIA Nsight Systems/Nsight Compute (or equivalent).
- Understanding of parallel computing and performance fundamentals (vectorization, threading, NUMA, memory bandwidth/latency).
- 5+ years relevant experience in GPU/HPC optimization; strong track record of delivered speedups and scaling improvements.
What You'll Be Doing:
- Profile, analyze, and optimize GPU-accelerated applications with emphasis on CUDA kernels, memory movement, concurrency, and end-to-end throughput.
- Drive performance improvements across the stack, including CUDA/C++ kernel optimization, GPU libraries usage, and multi-GPU/multi-node scaling.
- Build reproducible benchmarks, performance reports, and tuning recommendations.
- Develop and maintain reference implementations and patches to enable performance and portability.
- Support customer engagements from POCs to production, advising on best practices for deployment.
- Collaborate with internal teams to influence roadmap based on real customer requirements in Energy.
- Build internal libraries and reusable code for future NVIDIA products.
Nice to Haves:
- Leads performance reviews with customer stakeholders; creates reusable playbooks/reference designs.
- HPC experience with MPI, distributed systems, and multi-node performance tuning.
- Energy/HPC domain exposure including seismic processing, reservoir simulation, and power grid simulation/optimization workflows.
- Experience with CI/perf regression testing, containerized workflows (Docker/Apptainer), and schedulers (Slurm).
- Familiarity with AI workflows used alongside simulation for data prep, training/inference integration, and pipeline performance.
Perks and Benefits:
NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. If you're creative and autonomous, we want to hear from you!