Senior Software Engineer, GPU Code Generation for Machine Learning Inference

NVIDIA is seeking talented engineers to join the TensorRT team and advance high-performance AI inference on NVIDIA GPUs. Our team develops software that accelerates state-of-the-art generative AI models for local and enterprise deployments. We solve challenging problems in GPU code generation, just-in-time compilation, and inference optimization to deliver high performance across diverse models and hardware configurations.

As a Senior Software Engineer, you will develop advanced GPU inference software for TensorRT. We’re looking for engineers with strong experience in GPU optimization and compiler development who can translate a deep understanding of GPU execution into efficient code generation and measurable improvements in generative AI inference.

• Design and implement efficient GPU code generation techniques in TensorRT that balance generated code performance, compilation overhead, and startup latency.