Job Description
We are seeking a Senior Deep Learning Engineer to join our team at NVIDIA, a leader in AI and accelerated computing. This role is pivotal in optimizing performance across all layers of the hardware/software stack, from GPU architecture to deep learning frameworks, to achieve peak performance.
Responsibilities
- Implement deep learning models from various data domains such as CV, NLP/LLMs, ASR, TTS, RecSys, and others using multiple DL frameworks (PyTorch, JAX, TensorFlow, DGL, etc.).
- Develop and test new software features (e.g., Graph Compilation, reduced precision training) leveraging the latest hardware functionalities.
- Analyze, profile, and optimize deep learning workloads on cutting-edge hardware and software platforms.
- Collaborate with researchers and engineers across NVIDIA to enhance the design, usability, and performance of workloads.
- Lead best practices for building, testing, and releasing DL software.
Requirements
- 3+ years of experience in deep learning model implementation and software development.
- BSc, MS, or PhD in Computer Science, Computer Architecture, or a related technical field.
- Excellent Python programming skills and extensive knowledge of at least one DL framework (PyTorch, TensorFlow, JAX, MXNet).
- Strong problem-solving and analytical skills.
- Solid understanding of algorithms and deep learning fundamentals.
Preferred Qualifications
- Experience in performance measurements and profiling.
- Familiarity with containerization technologies such as Docker.
- GPU programming experience (CUDA or OpenCL) is a plus.
- Strong understanding of Linux environments.
- Knowledge and passion for DevOps/MLOps practices in deep learning product development.
Why NVIDIA?
NVIDIA is widely recognized as one of the most desirable employers in the technology sector. We are committed to fostering a diverse and inclusive workplace and are proud to be an equal opportunity employer. We do not discriminate based on race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.
Join us and be part of a team that is at the forefront of the AI revolution, working on projects that have a global impact.
Benefits Extracted with AI
- Equal opportunity employer
- Diversity and inclusion
Similar jobs
Last update: 23 minutes ago
Senior Deep Learning Performance Software Engineer
Senior role optimizing deep learning performance at NVIDIA, involving Python, HPC, and AI technologies.
Senior Full Stack Engineer, Deep Learning Algorithms
Join NVIDIA as a Senior Full Stack Engineer to build software for AI, focusing on deep learning algorithms and high-performance computing.
Machine Learning Engineer - LLM Fine-tuning and Performance
Join NVIDIA as a Machine Learning Engineer specializing in LLM fine-tuning and performance optimization. Work with cutting-edge ML technologies.
Senior Artificial Intelligence Algorithms Engineer
Senior AI Algorithms Engineer role focusing on AI/DL, data analytics, and machine learning in Santa Clara, CA.
Trustworthy AI Software Engineer
Join NVIDIA as a Trustworthy AI Software Engineer in Santa Clara, CA. Develop cutting-edge AI tools and models in a multidisciplinary team.
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.
Senior Backend Engineer, AI Platform - Robotics
Join NVIDIA as a Senior Backend Engineer to develop AI platforms for robotics. Work remotely with cutting-edge technology.
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.
Senior Software Engineer, AI Platform - Robotics
Senior Software Engineer needed for AI Robotics platform at NVIDIA, Santa Clara. Involves cloud platforms, Kubernetes, Python.
Senior C++ Computer Vision Engineer
Join a cutting-edge AI-DeepTech startup in Berlin as a Senior C++ Computer Vision Engineer. Work on world-class on-device AI technology.
Senior Software and System Architect
Join NVIDIA as a Senior Software and System Architect to lead cloud-networking and security solutions, focusing on cutting-edge technologies.
Senior Engineering Manager, Robotics and ML Applications
Senior Engineering Manager for Robotics & ML at NVIDIA, leading innovative projects in AI and robotics applications.
Senior Research Scientist, Multimodal Foundation Models and Robotics
Senior Research Scientist role at NVIDIA, focusing on AI and robotics, based in Santa Clara, CA.
Senior Full-Stack Software Engineer
Join NVIDIA as a Senior Full-Stack Software Engineer, working on cutting-edge web applications and infrastructure.
Senior Software & Cloud Architect
Join NVIDIA as a Senior Software & Cloud Architect to lead cloud-based orchestration and provisioning solutions.
Senior DevOps Engineer
Join NVIDIA as a Senior DevOps Engineer to enhance our Kubernetes platform and multi-cloud infrastructure.
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.
Senior Backend Engineer C++
Join DeepL as a Senior Backend Engineer C++ to design and maintain scalable backend services using C++ and AI technologies.
Senior AI/ML Engineer
Join AMD as a Senior AI/ML Engineer to develop and implement cutting-edge AI solutions using PyTorch and TensorFlow.
Senior Software Solution Engineer, Networking
Join NVIDIA as a Senior Software Solution Engineer in Networking, focusing on complex customer solutions and development.
Senior Distributed Systems Backend Engineer
Join NVIDIA as a Senior Distributed Systems Backend Engineer to shape the future of Cloud Gaming with GeForce NOW.
Software Engineering Intern - Math Libraries
Join NVIDIA as a Software Engineering Intern to develop GPU-accelerated libraries for AI and HPC applications.
Senior Applied AI Model Researcher
Join Ampere as a Senior Applied AI Model Researcher to lead AI model development and optimization in a remote role.
Senior Software Architect, Advanced Development
Join NVIDIA as a Senior Software Architect in Advanced Development, focusing on innovative solutions in network programmability and data centers.