Job Summary:
NVIDIA is a leading technology company specializing in GPU technology and AI solutions. They are seeking a Principal Software Engineer to design and implement a low-overhead GPU profiling service that provides insights for ML workloads, leading the architecture and delivery across system software and drivers.
Responsibilities:
โข Design the architecture for an Always-On profiling service, defining interfaces, data flows, and scalability guarantees for multi-process/GPU/node systems.
โข Drive low-overhead, high-reliability implementations in C/C++, including IPC/shared memory, and bounded CPU/memory budgets.
โข Lead end-to-end feature delivery spanning user-mode components, driver/platform layers, and performance counter/trace providers.
โข Establish profiling models that integrate with existing ML/AI workflows (e.g., PyTorch/XLA) to turn low-level signals into actionable insights.
โข Set technical direction for an engineering team; mentor engineers, drive technical planning to mitigate architectural risks, and align roadmaps across internal and external partners.
Qualifications:
Required:
โข BS or MS degree or equivalent experience in Computer Engineering, Computer Science, or related degree.
โข 15+ years of system-level C/C++ development, including concurrency, memory management, and performance engineering.
โข Proven experience designing and shipping production quality system software or drivers with strict reliability, observability, and performance constraints.
โข Demonstrated technical leadership: defining architecture and success metrics, and translating abstract product visions into actionable technical roadmaps with fast-paced, multidisciplinary teams.
โข Strong interpersonal, verbal, and written communication; able to influence across organizations and build trust with external collaborators.
Preferred:
โข Extensive experience with profiling/tracing stacks for CPU/GPU (e.g., CUPTI, Nsight, performance counters, event correlation) and debugging highly concurrent systems.
โข Deep hands-on knowledge of CUDA and GPU architecture, including runtime/driver APIs, CUDA streams/graphs, and kernel behavior.
โข Track record building continuous, always-on, or multi-client profiling systems designed for predictable overhead at scale.
โข Hands-on experience tuning ML training/inference loops based on deep profiling analysis, with familiarity in ML ecosystems (e.g., PyTorch, JAX) and correlating application events with GPU metrics to translate data into actionable performance insights (e.g., bottleneck triage, compute vs. memory bound).
โข Experience with user-mode driver development and integration within platform security and permissions models.
Company:
NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI. Founded in 1993, the company is headquartered in Santa Clara, USA, with a team of 10001+ employees. The company is currently Late Stage.