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
Senior Deep Learning Engineer
Join NVIDIA as a Senior Deep Learning Engineer to optimize AI performance using PyTorch, TensorFlow, and more in Berlin.
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.
Senior Artificial Intelligence Algorithms Engineer
Senior AI Algorithms Engineer role focusing on AI/DL, data analytics, and machine learning in Santa Clara, CA.
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 Software Engineer, AI Platform - Robotics
Senior Software Engineer needed for AI Robotics platform at NVIDIA, Santa Clara. Involves cloud platforms, Kubernetes, Python.
Senior Software Solution Engineer, Networking
Join NVIDIA as a Senior Software Solution Engineer in Networking, focusing on complex customer solutions and development.
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.
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 Software Architect, Advanced Development
Join NVIDIA as a Senior Software Architect in Advanced Development, focusing on innovative solutions in network programmability and data centers.
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.
Senior Software Engineer - HPC
Senior Software Engineer for HPC at NVIDIA in Westford, MA. Design and improve high-performance computing systems.
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.
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 Software & Cloud Architect
Join NVIDIA as a Senior Software & Cloud Architect to lead cloud-based orchestration and provisioning solutions.
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 Full-Stack Web Applications Software Engineer
Join NVIDIA as a Senior Full-Stack Web Applications Software Engineer. Work on scalable web services and infrastructure.
Senior Software Architect – Data Center Platform Simulation and Virtualization
Join NVIDIA as a Senior Software Architect focusing on data center platform simulation and virtualization.
Senior Full-Stack Software Engineer
Join NVIDIA as a Senior Full-Stack Software Engineer, working on cutting-edge web applications and infrastructure.
Senior Engineering Manager, Robotics and ML Applications
Senior Engineering Manager for Robotics & ML at NVIDIA, leading innovative projects in AI and robotics applications.
Senior DevOps Engineer
Join NVIDIA as a Senior DevOps Engineer to enhance our Kubernetes platform and multi-cloud infrastructure.
Senior Research Scientist, Multimodal Foundation Models and Robotics
Senior Research Scientist role at NVIDIA, focusing on AI and robotics, based in Santa Clara, CA.
Software Engineering Intern - Math Libraries
Join NVIDIA as a Software Engineering Intern to develop GPU-accelerated libraries for AI and HPC applications.
Senior Production SRE Engineer - Storage
Join NVIDIA as a Senior Production SRE Engineer - Storage, ensuring reliability of GPU cloud services with cutting-edge technologies.
Software Engineering Intern - NVIDIA Ignite Program
Join NVIDIA's 2025 Ignite Software Engineering Internships for hands-on experience in cutting-edge technology.
