Principal Software Engineer, Visual AI

Key Skills

Aartificial intelligence, C++, Computer Vision (CV), CUDA kernels, Debugging, Docker, Generative AI, Helm, Kubernetes-K8s, Machine learning techniques, Python Programming, PyTorch

Job Description

NVIDIA is a world-leader in artificial intelligence, computer vision, and accelerated computing. Our technology is a driving force in the advancement of visual AI from data centers to cars, robots, smart buildings, medical devices, and more. As models grow in capability, data scale, and generality, developers are faced with complex and expensive challenges from preparing hundreds of petabytes of data to validation of open world behaviors. Our team is building the accelerate software ecosystem to help developers through these barriers and to enable the next wave of advances in computer vision, multi-modal models, and GenAI.

We are now looking for an innovative technical leader to drive the capability of developer facing products for visual search, curation, 3d reconstruction, and more. The right candidate will combine curiosity for ground breaking research, creativity to identify developer focused applications of a range of computer vision and GenAI techniques, strong rapid prototyping skills, desire to teach and share technical vision, and a strong production sensibility. Looking to make computer vision ubiquitous, save the world billions of dollars and drive data center efficiency? This could be a phenomenal opportunity!

What you'll be doing:

• Contributing to our technical strategy for visual AI developer software through insight into key developer challenges and innovative ways to mitigate those challenges with accelerated computing.

• Invent and prototype new features, tools, workflows, and services that help AI developers advance the state of the art faster and more cost-effectively.

• Drive new capabilities to production services that will run on hundreds or thousands of GPUs including profiling and optimizing code, building tests, deploying microservices, debugging quality gaps, fine tuning and optimize models, automating workflows, creating metrics and trackers.

• Lead major projects including feature and architectural definition, project planning and estimation, execution, and mentorship for teammates.

• Collaborate with our research teams to adapt their work and to influence research direction

• Provide technical expertise in deep-learning based computer vision to teams across NVIDIA and support critical initiatives in this area

• Demonstrate consistently high levels of creativity resulting in patents, product differentiation and advances in visual AI.

What we need to see:

• MS or equivalent experience in Computer Science, Computer Engineering, or related field.

• Outstanding technical breadth across areas such as generative AI, visual understanding, and 3d reconstruction

• Proficiency with Python and C++. Basic familiarity with GPU programming, e.g. in CUDA.

• 10+ year track record of converting research ideas into production software, leading development activities including algorithm selection, implementation, data collection, and validation.

• Innovation resulting in outstanding products, patents, and publications.

Ways to Stand Out from the Crowd:

• Proficiency with common cloud software such as Helm Charts, Milvus, ELK, Docker, etc.

• Strong software architecture skills

• Experience building developer centric tools and services

We are widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and talented people in the world working for us. If you're creative and autonomous with a real passion for technology we want to hear from you. We are an equal opportunity employer and value diversity at our company.

The base salary range is 272,000 USD - 425,500 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. NVIDIA accepts applications on an ongoing basis.

NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.