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.
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 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.
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 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 Full-Stack Software Engineer
Join NVIDIA as a Senior Full-Stack Software Engineer, working on cutting-edge web applications and infrastructure.
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 Software Engineer - HPC
Senior Software Engineer for HPC at NVIDIA in Westford, MA. Design and improve high-performance computing systems.
Information Retrieval Algorithm Engineer
Join Huawei as an Information Retrieval Algorithm Engineer to develop cutting-edge AI technologies in Amsterdam.
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 Engineer (C++)
Join Nebius as a Senior Software Engineer (C++) to develop reliable cloud services in a hybrid work environment.
Senior Software & Cloud Architect
Join NVIDIA as a Senior Software & Cloud Architect to lead cloud-based orchestration and provisioning solutions.
Senior Fullstack Developer for AI-Driven Mission Technologies
Seeking a Senior Fullstack Developer for AI-driven mission technologies, focusing on Java, JavaScript, Python, and C++. Remote work available.
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 Engineering Manager, Robotics and ML Applications
Senior Engineering Manager for Robotics & ML at NVIDIA, leading innovative projects in AI and robotics applications.