Senior Deep Learning Engineer, Visual Generative AI

Senior Deep Learning Engineer, Visual Generative AI

We are looking for a Senior DL Algorithms Engineer with experience optimizing and deploying Deep Learning models focusing on Diffusion Models and Vision-Language Models (VLMs) in production environments. In this role, you will optimize and deploy deep learning models for efficient and fast inference across diverse GPU platforms, particularly for Visual Generative AI applications.

Join the team building software used by the entire world. Work with world-class research scientists, software engineers, and hardware specialists to bring cutting-edge AI models from prototype to production.

Responsibilities

• Optimize deep learning models for low-latency, high-throughput inference, with a focus on Diffusion models for Visual Generative AI applications.
• Convert, deploy, and optimize models for efficient inference using frameworks such as TensorRT, TensorRT-LLM, and vLLM.
• Understand, analyze, profile, and optimize the performance of deep learning workloads on state-of-the-art NVIDIA GPU hardware and software platforms.
• Collaborate with internal and partner research scientists and software engineers to ensure seamless integration of cutting-edge AI models from training to deployment.
• Contribute to the development of automation and tooling for NVIDIA Inference Microservices (NIMs) and inference optimization, including creating automated benchmarks to track performance regressions.

What We Need To See

• 3+ years of experience in DL model implementation and software development.
• BSc, MS or PhD degree in Computer Science, Computer Architecture or related technical field.
• Extensive knowledge of at least one DL Framework (PyTorch, JAX, TensorFlow) with practical experience in PyTorch required.
• Deep understanding of transformer architectures, attention mechanisms, Visual Generative AI foundational models architectures (e.g., U-Net, DiT) and inference bottlenecks.
• Excellent Python programming skills.
• Strong problem solving and analytical skills.
• Algorithms and DL fundamentals.
• Docker containerization fundamentals.

Ways To Stand Out From The Crowd

• Experience in performance measurements and profiling.
• Hands-on experience with model optimization and serving frameworks, such as: TensorRT, TensorRT-LLM, vLLM, SGLang, and ONNX.
• Deep understanding of distributed systems for large-scale model inference and serving.
• Experience with extending and leveraging open-source tools for Visual Generative AI workflow creation.
• Familiarity with the latest trends in Visual Generative AI for content creation.

NVIDIA is committed to diversity and equal opportunity. We value a diverse workforce and do not discriminate on the basis of race, religion, color, national origin, gender, sexual orientation, age, marital status, veteran status, or disability status.