Deep Learning Software Engineer

NVIDIA is searching for Deep Learning algorithms architect to develop artificial intelligence (AI), computer vision algorithms and applications for our Metropolis for Factories and Manufacturing platforms. Artificial Intelligence is transforming how we collect, inspect, and analyze different kinds of sensor data that impacts everything from manufacturing automation, warehouse management, product inspections to safety workflows. NVIDIA Metropolis is leading this AI revolution, providing the tools, technologies, and expertise to meet every challenge with smarter, faster applications.

This challenging role will require someone who deeply understands and can architect algorithms with Large Language and Multi-modal (LLM/LMM) Foundation models to advance the application of artificial intelligence and machine learning to the Manufacturing AI market. Practical experience in the use and the building of Computer Vision algorithms, models and tools will be critical.

What You’ll Be Doing:

• Architect, analyze, develop, and prototype key deep learning algorithms and solutions as a core member of our growing software team.

• Optimize deep learning models for deployment using techniques like model distillation, quantization, pruning, and others to ensure the highest efficiency across platforms.

• Develop and implement processes for model conversion and deployment, targeting both on-premises and cloud environments.

• Collaborate with diverse software, research, and hardware teams across geographies to analyze the interplay of hardware and software architectures, solving critical problems and envisioning future applications.

• Work closely with customers and their third-party software providers to provide support and enable successful engagements. This will also involve collaboration with Product Management, Marketing, and Developer Technology teams.

• Lead the design and implementation of complex AI projects, providing technical guidance and support, and mentoring junior engineers to cultivate expertise in advanced AI methodologies.

What We Need To See:

• 2 years or more of working experience and an MS or PhD in Computer Science, Computer Engineering, Electrical Engineering, or a related field with a focus on Deep Learning, Machine Learning, and Computer Vision.

• Expertise in deep learning model optimization, including knowledge and experience in model distillation, quantization, pruning, and other techniques that enhance computational efficiency.

• Demonstrable experience with model conversion and deployment processes, from frameworks like TensorFlow and PyTorch into efficient runtime formats.

• Experience with advanced algorithms, including zero/few-shot learning, self-supervised and unsupervised learning techniques, domain adaptation methods like PEFT (Parameter-Efficient Fine-Tuning), and generative AI models for synthetic dataset creation.

• Proficiency in working with deep learning frameworks such as TensorFlow and PyTorch, Strong programming skills in Python and/or C++,  and Experience developing integrated AI solutions.

• Proven ability to lead projects, manage timelines, and deliver results.

• Strong communication skills and ability to work in a collaborative environment.

Ways To Stand Out From The Crowd:

• Experience in optimizing models for deployment using advanced techniques such as quantization and pruning, and familiarity with large-scale data processing frameworks.

• Proven experience in algorithm development for data analytics, especially with LLMs and Multi-Modal Foundation models. Experience working and curation with multi-modal datasets.

• Knowledge of software development best practices, including version control, code review, and documentation.

• Track record of contributions to academic publications in the field of AI/ML or open-source projects.

NVIDIA is widely considered to be one of the technology world’s most desirable employers. We have some of the most forward-thinking and hardworking people in the world working for us. Are you a creative and autonomous data scientist who loves challenges? Do you have a genuine passion for advancing the state of AI & machine learning across a variety of industries? If so, we want to hear from you. Come and join our Metropolis / team where you'll help build our real-time, cost-effective computing platform driving our success in this exciting and quickly growing field.

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.