Senior Deep Learning Engineer

September 17

Apply Now

Receive Emails with Similar Jobs

NVIDIA

WebsiteLinkedInAll Job Openings

GPU-accelerated computing • artificial intelligence • deep learning • virtual reality • gaming

10,000+

Description

• Implement deep learning models from multiple data domains (CV, NLP/LLMs, ASR, TTS, RecSys and others) in multiple DL frameworks (PyT, JAX, TF2, DGL and others) • 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

Requirements

• 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) • Strong problem solving and analytical skills • Algorithms and DL fundamentals • Experience in performance measurements and profiling • Experience with containerization technologies such as Docker • GPU programming experience (CUDA or OpenCL) is a plus but not required. • Solid understanding of Linux environments • Knowledge and love for DevOps/MLOps practices for Deep Learning-based product’s development.