Software Engineer - Deep Learning

NVIDIA is known as “the AI computing company.” Come, join our Deep Learning team, where you can help build the real-time, cost-effective computing platform driving our success in this exciting and quickly growing field. We are currently seeking an experienced senior software engineer with strong Deep Learning and System Programming fundamentals coupled with robust C / C++ skills to contribute to the development of NVIDIA Maxine Audio - a comprehensive suite of SDKs, Microservices and APIs that enable AI-driven features for video conferencing, content creation and gaming.

What you’ll be doing :

Architect and lead development of next-generation Deep Learning and multimedia algorithms for processing of speech and audio applications.

Train Speech Enhancement models, assess them for quality, performance, and finetune them.

Analyze model accuracy and bias and recommend the next course of action & Improvements.

Improve processes for speech data processing, augmentation, filtering & training sets preparation.

Optimize algorithms for optimal performance on the GPU tensor cores

Collaborate with various teams to drive an end to end workflow from data curation and training to performance optimization and deployment

Influence strategic decisions in the team and product roadmap

Partner with system software engineers and validation teams to build and ship production-quality code.

What we need to see :

PH.D. / MS in Computer Science or a closely related engineering field with 3+ years of relevant experience

Strong background in Deep Learning including model design, pruning & performance optimization, transfer learning etc

4+ years of experience of leading cross-module projects and taking them to productization

Strong software engineering background with proficiency in C or C++

Hands-on expertise with PyTorch, TensorRT, CuDNN and one or more Deep Learning frameworks (Tensorflow, Keras etc)

Familiarity / expertize with various cloud frameworks e.g. AWS, GCP, Azure is a big plus

CUDA programming experience is a plus

Excellent communication and collaboration skills

Self-motivated and able to find creative practical solutions to problems

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.