Windows AI Software Engineer

Details

At NVIDIA, we’re tapping into the unlimited potential of AI to define the next era of computing. An era in which our GPU acts as the brains of computers, robots, and self-driving cars that can understand the world. Doing what’s never been done before takes vision, innovation, and the world’s best talent. As an NVIDIAN, you’ll be immersed in a diverse, supportive environment where everyone is inspired to do their best work. Come join the team and see how you can make a lasting impact on the world!

Responsibilities

• Partnering with NVIDIA software, research, architecture, and product teams, aligning strategies and technical needs for fostering the ecosystem of AI on a Windows RTX PC.
• Performing in-depth analysis and optimization of AI models, AI frameworks, data processing pipelines, and inference backends to ensure the best performance on current and next-generation GPU architectures.
• Identifying and implementing compute and memory optimizations across the full AI inference stack on RTX Windows PC.
• Developing model compression and fine-tuning techniques to reduce resource consumption and improve performance, enabling efficient deployment and better user experience.
• Designing and implementing an optimized framework for running AI NPCs in gaming applications as part of the NVIDIA ACE Platform.
• Collaborating with Microsoft to drive advancements in APIs, AI frameworks, and platforms for developing and deploying AI inferencing applications.
• Ensuring the effective deployment of directed tests through collaboration with the automation team, thereby ensuring the robustness of automated testing.

Requirements

• Bachelor's, Master's, or PhD in Computer Science, Software Engineering, Mathematics, or a related field (or equivalent experience).
• 5+ years experience in a related software position.
• 2+ years of experience in AI inferencing pipelines and applications using ML/DL frameworks like PyTorch, ONNX Runtime, DirectML preferred.
• Excellent C++ programming and debugging skills with a strong understanding of data structures and algorithms.
• Strong analytical and problem-solving abilities, with the capacity to multitask effectively in a dynamic environment.
• Outstanding written and oral communication skills, enabling effective collaboration with management and engineering teams.

Benefits

Widely considered to be one of the technology world’s most desirable employers, NVIDIA offers highly competitive salaries and a comprehensive benefits package. The base salary range is 148,000 USD - 230,000 USD. Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. You will also be eligible for equity and benefits.