Senior System Software Engineer, DirectML - GPU

NVIDIA

NVIDIA

Software Engineering
Multiple locations
Posted on Saturday, June 29, 2024

NVIDIA is a pioneering company that revolutionized the computing world with its GPU technology, driving the growth of the PC gaming market and redefining modern computer graphics. In recent times, GPU deep learning has propelled AI into the next era of computing, with GPUs serving as the brains behind AI-powered applications across a wide range of industries, including high performance computing, robotics, autonomous vehicles, and more. Today, NVIDIA is renowned as the "AI computing company."

We are currently seeking a highly motivated Senior Systems Software Engineer to join our dynamic DirectML driver team. If you are enthusiastic about working across all layers of the software stack, encompassing deep learning frameworks, drivers and GPU architecture, and if the prospect of implementing novel features and achieving peak performance excites you, we eagerly await hearing from you!. This role allows you to directly influence the hardware and software roadmap of a fast-expanding AI technology company, making a significant impact on the experience of gamers and other PC users as they relish the latest AI innovations.

What you’ll be doing:

  • Collaborate with cross-functional teams to understand requirements and contribute to the design and architecture of NVIDIA DirectML drivers.

  • Develop and implement high-quality, efficient, and reliable NVIDIA DirectML drivers following the best software development practices adhering to industry standards.

  • Identify and resolve performance bottlenecks for cutting-edge deep learning models in computer vision, generative AI, and other related fields.

  • Debug and resolve driver-related issues to ensure smooth functionality across multiple GPU architectures and platforms.

  • Collaborate with Microsoft to drive the advancements in ONNX and DML runtimes and ensure seamless integration with NVIDIA DirectML driver.

  • Build and contribute to performance profiling tools that improves the efficiency of the performance analysis workflow.

  • Ensure the effective deployment of essential directed tests and models through collaboration with the automation team, thereby ensuring the robustness of automated testing.

What we need to see:

  • Bachelor's, Master's, or PhD in Electrical Engineering, Computer Science/Engineering, Mathematics, Physics, or a related field (or equivalent experience).

  • 5+ years of industry experience with proficiency in developing and optimizing DirectML drivers or similar graphics and machine learning drivers such as DirectX, Vulkan, or CUDA.

  • Familiarity with GPU hardware architecture, shader programming, and parallel processing techniques.

  • Excellent C++ programming and debugging skills with a strong understanding of data structures and algorithms.

  • Familiarity with deep learning frameworks such as PyTorch and TensorFlow and exposure to large language models and transformer architectures.

  • Strong analytical and problem-solving abilities, with the capacity to multitask effectively in a multifaceted environment.

  • Outstanding written and oral communication skills, enabling effective collaboration with management and engineering teams.

If you are ready to make a significant impact in the world of AI and GPU technology, join us at NVIDIA, where you will have the opportunity to craft the future of computing while working with a passionate and talented team