Expoint - all jobs in one place

Finding the best job has never been easier

Limitless High-tech career opportunities - Expoint

Nvidia Senior System Software Engineer DirectML - GPU 
India, Maharashtra, Pune 
58259436

18.08.2024

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.