Senior System Software Engineer, DirectML - GPU
India, Pune
NVIDIA
NVIDIA erfindet den Grafikprozessor und fördert Fortschritte in den Bereichen KI, HPC, Gaming, kreatives Design, autonome Fahrzeuge und Robotik.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
* Salary range is an estimate based on our AI, ML, Data Science Salary Index 💰
Tags: Architecture Computer Science Computer Vision CUDA Deep Learning Engineering Generative AI GPU HPC LLMs Machine Learning Mathematics ONNX PhD Physics PyTorch Robotics TensorFlow Testing Vulkan
Perks/benefits: Career development Startup environment
More jobs like this
Explore more AI, ML, Data Science career opportunities
Find even more open roles in Artificial Intelligence (AI), Machine Learning (ML), Natural Language Processing (NLP), Computer Vision (CV), Data Engineering, Data Analytics, Big Data, and Data Science in general - ordered by popularity of job title or skills, toolset and products used - below.
- Open Research Scientist jobs
- Open Data Science Manager jobs
- Open Data Engineer II jobs
- Open Principal Data Scientist jobs
- Open Business Data Analyst jobs
- Open BI Analyst jobs
- Open Sr Data Engineer jobs
- Open Data Scientist II jobs
- Open Business Intelligence Engineer jobs
- Open Data Science Intern jobs
- Open Sr. Data Scientist jobs
- Open Senior Business Intelligence Analyst jobs
- Open Software Engineer, Machine Learning jobs
- Open Lead Data Analyst jobs
- Open Azure Data Engineer jobs
- Open Junior Data Scientist jobs
- Open MLOps Engineer jobs
- Open Data Analytics Engineer jobs
- Open Manager, Data Engineering jobs
- Open Marketing Data Analyst jobs
- Open Data Engineer III jobs
- Open Junior Data Engineer jobs
- Open Data Engineering Manager jobs
- Open Data Analyst II jobs
- Open ETL Developer jobs
- Open Data quality-related jobs
- Open Data management-related jobs
- Open Privacy-related jobs
- Open Excel-related jobs
- Open ML models-related jobs
- Open Data pipelines-related jobs
- Open APIs-related jobs
- Open PhD-related jobs
- Open PyTorch-related jobs
- Open LLMs-related jobs
- Open Finance-related jobs
- Open Data visualization-related jobs
- Open Consulting-related jobs
- Open TensorFlow-related jobs
- Open Deep Learning-related jobs
- Open Business Intelligence-related jobs
- Open Generative AI-related jobs
- Open CI/CD-related jobs
- Open NLP-related jobs
- Open Data governance-related jobs
- Open DevOps-related jobs
- Open Kubernetes-related jobs
- Open Git-related jobs
- Open Hadoop-related jobs
- Open Docker-related jobs