Senior Software Engineer - Deep Learning

Senior Software Engineer - Deep Learning

NVIDIA has been transforming computer graphics and accelerated computing for more than 25 years. In the AI era, it’s a unique legacy of innovation that’s fueled by great technology and amazing people. NVIDIA Maxine is a developer platform for creating and deploying AI features for real-time video communication and content creation. Maxine provides state-of-the-art AI models for deploying video/audio enhancement and augmented reality features. Maxine also offers end-to-end pipelines of multiple low latency AI effects accelerated on NVIDIA GPUs.

We are now looking for outstanding engineers to join the NVIDIA Maxine AI team. You will work alongside brilliant engineers on core technologies to solve ambitious computer vision and deep learning problems, especially building and optimizing real-time AI solutions that could run anywhere on cloud or premise.

What You’ll Be Doing:

• Develop highly efficient and low-cost AI models and algorithms for computer vision and video AI.
• Optimize the performance, latency, and power consumption of AI models on low power processors for deep learning acceleration.
• Deploy deep learning models and optimize the inference stack for real-time performance.
• Deliver the benefits of NVIDIA’s latest hardware and platform software innovations to the Deep Learning team.
• Closely collaborate with different deep learning software and hardware teams across NVIDIA to influence roadmaps and deliver solutions.

What We Need To See:

• Strong experience in building and optimizing AI model architectures for fast inference such as model pruning, distillation, post-quantization, and quantization aware training.
• Experience with analyzing and fine-tuning deep learning pipeline performance.
• Experience with building real-time AI models for laptop and cloud use cases.
• 5+ years of relevant engineering or research background in deep learning and/or computer vision.
• Hands-on development skills using deep learning libraries and frameworks such as PyTorch, TensorFlow, ONNX, TensorRT, Triton, and other neural processing SDKs.
• Collaboration ability to define project scope and roadmap together with the team while independently driving development efforts with strong self-motivation.
• MS or PhD in Computer Science, Computer Engineering, or closely related field or equivalent experience.

Ways to stand out from the crowd:

• Experience with AI inference accelerating hardware and building/optimizing models on them.
• Background with performance and latency analysis, profiling, and tuning of AI workloads.
• Experience with CUDA programming, as well as a real passion for optimizing AI system performance.
• Experience of building platforms for computer vision such as real-time tracking of human face, gaze, and body, as well as avatar animation and modeling.

NVIDIA is widely considered to be one of the technology world’s most desirable employers. As a part of NVIDIA, we have the opportunity to influence the future with your vision and expertise. Are you creative? We have some of the most forward-thinking and hardworking people in the world working for us, and due to unprecedented growth, our exclusive engineering teams are rapidly growing. If you're a creative and autonomous engineer with a real passion for technology, we want to hear from you.

The base salary range is 148,000 USD - 276,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. 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.