NVIDIA logo

Senior Deep Learning Performance Software Engineer

NVIDIA

Job Overview

We are seeking a Senior Deep Learning Performance Software Engineer to join our team at NVIDIA, a leader in accelerated computing and AI technologies. This role is crucial for optimizing deep learning training, inference, and NVIDIA AI Services to enhance performance across all layers of the hardware/software stack.

Responsibilities

  • Implement deep learning models across various data domains such as CV, NLP/LLMs, ASR, TTS, RecSys in multiple deep learning frameworks (PyTorch, JAX, TensorFlow 2, DGL, etc.).
  • Develop and test new software features like Graph Compilation and reduced precision training, leveraging the latest hardware functionalities.
  • Analyze, profile, and optimize deep learning workloads on state-of-the-art hardware and software platforms.
  • Collaborate with researchers and engineers across NVIDIA to improve workload design, usability, and performance.
  • Lead best practices for building, testing, and releasing deep learning software.

Qualifications

  • 5+ years of experience in deep learning model implementation and software development.
  • Bachelor's, Master's, or PhD degree in Computer Science, Computer Architecture, Mathematics, Physics, or a related technical field.
  • Proficient in Python programming and knowledgeable in at least one deep learning framework.
  • Strong problem-solving and analytical skills.

Preferred Skills

  • Experience in performance measurements and profiling.
  • Familiarity with running large-scale workloads in HPC clusters.
  • Knowledge of DevOps/MLOps practices for deep learning-based product development.
  • Understanding of Linux environments and containerization technologies like Docker.
  • GPU programming experience (CUDA or OpenCL) is a plus.

Why NVIDIA?

Join a company that is at the forefront of the AI revolution, with some of the most brilliant minds in the technology world. We value diversity and are committed to creating an inclusive environment for all employees. This is an opportunity to work on powerful, enterprise-grade GPU clusters and impact the hardware and software roadmap in a significant way.

Benefits
Extracted with AI

  • Remote work flexibility

Similar jobs

Last update: 23 minutes ago

NVIDIA logo
NVIDIA

Senior Full Stack Web Software Engineer

Join NVIDIA as a Senior Full Stack Web Software Engineer to build AI-assisted developer tools using React.js and TypeScript.

NVIDIA logo
NVIDIA

Deep Learning Computer Architecture Intern

Join NVIDIA as a Deep Learning Computer Architecture Intern. Work on cutting-edge AI projects with a leading company in accelerated computing.

NVIDIA logo
NVIDIA

Artificial Intelligence and Deep Learning Intern

Join NVIDIA's 2025 AI and Deep Learning Internship to work on cutting-edge projects in AI, robotics, and more.

NVIDIA logo
NVIDIA

Software Engineering Intern

Join NVIDIA as a Software Engineering Intern in 2025. Work remotely on AI, cloud, and data science projects. Enhance your skills in a diverse environment.

NVIDIA logo
NVIDIA

Senior Deep Learning Engineer

Join NVIDIA as a Senior Deep Learning Engineer to optimize AI performance using PyTorch, TensorFlow, and more in Berlin.

NielsenIQ logo
NielsenIQ

Senior Machine Learning Engineer

Join NIQ as a Senior ML Engineer to develop and implement AI models using Python, PyTorch, and Azure in a hybrid work environment.

PhysicsX logo
PhysicsX

Machine Learning Scientist

Join PhysicsX as a Machine Learning Scientist to develop innovative models using deep learning for physics simulations.

Nike logo
Nike

Senior Machine Learning Engineer

Join Nike as a Senior Machine Learning Engineer to develop and optimize ML algorithms for innovative applications.

Niantic, Inc. logo
Niantic, Inc.

Senior Software Engineer, Machine Learning

Join Niantic as a Senior Software Engineer in Machine Learning to enhance products using generative AI technologies.

Atypon logo
Atypon

Senior Machine Learning Engineer

Join Atypon as a Senior ML Engineer to develop AI solutions in NLP, deep learning, and MLOps. Remote position in Athens.

Leonardo.Ai logo
Leonardo.Ai

Mid-Level AI Researcher

Join Leonardo.Ai as a Mid-Level AI Researcher to develop and refine AI models, focusing on model training and optimization.

NVIDIA logo
NVIDIA

Senior DevOps Engineer

Join NVIDIA as a Senior DevOps Engineer to enhance our Kubernetes platform and multi-cloud infrastructure.

Waabi logo
Waabi

Remote Software Engineer

Join Waabi as a Remote Software Engineer to develop cutting-edge self-driving technology. Work with AI, Python, C++, and more.

NPO logo
NPO

Cloud Data Engineer

Join NPO as a Cloud Data Engineer to enhance data platforms using GCP, Python, and more. Flexible hours and growth opportunities.

Inclusively logo
Inclusively

Senior Software Engineer, Machine Learning

Join as a Senior Software Engineer in Machine Learning, working remotely to build ML-driven products for user engagement.

SSi People logo
SSi People

Senior Machine Learning Engineer

Join as a Senior Machine Learning Engineer to design and deploy advanced ML solutions using Python, Spark, and cloud platforms. Remote work opportunity.

NICE logo
NICE

Data Scientist with NLP and Machine Learning Expertise

Join NICE as a Data Scientist to develop NLP algorithms and models, enhancing contact center solutions.

ASML logo
ASML

Machine Learning Mathematics Engineer

Join ASML as a Machine Learning Mathematics Engineer to develop optical metrology solutions using machine learning and optimization algorithms.

Intel Corporation logo
Intel Corporation

AI Software Development Engineer

Join Intel as an AI Software Development Engineer to develop and deploy AI applications, enhancing engineering productivity.

3Pillar logo
3Pillar

Senior AI/ML Engineer

Join 3Pillar as a Senior AI/ML Engineer to develop innovative AI solutions in a remote, global team.

Parrot logo
Parrot

Senior Machine Learning Engineer - NLP

Join Parrot as a Senior Machine Learning Engineer focusing on NLP to develop cutting-edge AI solutions for legal tech.

Nebius AI logo
Nebius AI

MLOps Engagement Engineer

Join Nebius AI as an MLOps Engagement Engineer to design and optimize ML workflows using Kubernetes, Docker, and Slurm.

Tesla logo
Tesla

Internship Software Engineer - Machine Learning Infrastructure

Join Tesla as an Internship Software Engineer in Machine Learning Infrastructure. Work on AI infrastructure and neural network scaling.

micro1 logo
micro1

Machine Learning Engineer with AI/ML Experience

Join us as a Machine Learning Engineer to develop AI/ML models and applications. Work remotely with top-tier companies.