[Remote] Senior Deep Learning Engineer, Visual Generative AI at NVIDIA

Candidates should possess a Bachelor's, Master's, or PhD degree in Computer Science, Computer Architecture, or a related technical field, along with 3+ years of experience in Deep Learning model implementation and software development. Extensive knowledge of at least one Deep Learning framework, with practical experience in PyTorch, is required, as is a deep understanding of transformer architectures, attention mechanisms, Visual Generative AI foundational model architectures (e.g., U-Net, DiT), and inference bottlenecks. Excellent Python programming skills, strong problem-solving and analytical skills, fundamental knowledge of algorithms and Deep Learning, and familiarity with Docker containerization are also necessary. Experience with performance measurements, profiling, model optimization and serving frameworks like TensorRT, TensorRT-LLM, vLLM, SGLang, and ONNX, as well as a deep understanding of distributed systems for large-scale model inference and serving, will help candidates stand out.