NVIDIA is looking for a Senior Deep Learning Engineer, Deep Learning Algorithms to maximize the performance of our AI platforms. You'll focus on performance analysis and optimization, impacting hardware and software roadmaps across Deep Learning training, inference, and NVIDIA AI Services.
What You'll Do
- Implement deep learning models from multiple data domains (CV, NLP/LLMs, ASR, TTS, RecSys) in multiple DL frameworks (PyTorch, JAX, TensorFlow2, DGL).
- Implement and test new SW features (Graph Compilation, reduced precision training) that use the most recent HW functionalities.
- Analyze, profile, and optimize deep learning workloads on state-of-the-art hardware and software platforms.
- Collaborate with researchers and engineers across NVIDIA, providing guidance on improving the design, usability and performance of workloads.
- Lead best-practices for building, testing, and releasing DL software.
- Contribute to creation of large scale benchmarking system, capable of testing thousands of models on vast diversity of hardware and software stacks.
What We're Looking For
- 3+ years of experience in DL model implementation and SW Development.
- BSc, MS or PhD degree in Computer Science, Computer Architecture or related technical field.
- Excellent Python programming skills.
- Extensive knowledge of at least one DL Framework (PyTorch, TensorFlow, JAX, MxNet) with practical experience in PyTorch required.
- Strong problem solving and analytical skills.
- Algorithms and DL fundamentals.
- Docker containerization fundamentals.
Nice to Have
- Experience in performance measurements and profiling.
- Experience with containerization technologies such as Docker.
- GPU programming experience (CUDA or OpenCL).
- Knowledge and love for DevOps/MLOps practices for Deep Learning-based product’s development.
- Experience with CI systems (preferably GitLab).
Technical Stack
- Languages & Frameworks: Python, PyTorch, TensorFlow, JAX, MxNet, DGL
- Infrastructure & Tools: Docker, CUDA, OpenCL, GitLab
Team & Environment
Work with world class software engineers and collaborate with researchers and engineers across NVIDIA.
We do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.
