DirectML Framework Engineer

• Design, develop, and maintain DirectML API implementation and User-Mode Drivers for GPUs, CPUs and AI Accelerators.
• Strong understanding of Windows Driver Model (WDM), Device Driver Interface (DDI), User-Mode Drivers, Kernel-Mode drivers; Microsoft Compute Driver Model (MCDM), Windows Display Driver Model (WDDM).
• Collaborate with hardware vendors to optimize driver performance for various machine-learning frameworks like TensorFlow, PyTorch, and ONNX.
• Ensure seamless integration of DirectML drivers with Windows, focusing on performance, stability, and feature support.
• Conduct thorough testing, validation, and performance benchmarking of DirectML UMD across various hardware configurations.
• Implement support for new hardware features in DirectML drivers and adapt to evolving machine-learning algorithms.
• Contribute to DirectML SDK development, providing guidance on driver APIs, performance, and best practices.
• Monitor and ensure driver quality through robust testing and validation of DirectML workloads on supported hardware.

Required Skills & Qualifications:

• Strong proficiency in C/C++ and experience with DirectML API implementation and User-Mode Drivers.
• Deep understanding of GPU, CPU and AI Accelerator architectures, with hands-on experience in performance tuning for ML workloads.
• Familiarity with Microsoft’s DirectX and Windows Driver Model (WDM), Device Driver Interface (DDI), User-Mode Drivers.
• Strong debugging and profiling skills, with experience using tools
• Ability to work in a fast-paced, collaborative environment, with excellent communication and teamwork skills.

Experience – 4 to 10 Years

Qualifications

• B.Tech/B.E/M.Tech/M.E