Senior Software Engineer, Deep Learning - MLIR TRT

## Responsibilities - Developing compiler technologies to accelerate deep learning inference on NVIDIA hardware platforms for Physical AI. - Working across a wide range of abstractions from model fine-tuning and quantization to low-level kernel development and performance optimization. - Develop workflows that let users leverage frameworks (e.g. PyTorch, JAX) and compiler technologies tools (e.g. MLIR, Triton) without forgoing performance. - Work with customers to help accelerate their workloads on NVIDIA platforms. - Stay up to date with the latest research and innovations in deep learning, implement and experiment with new insights to improve NVIDIA's Physical AI DNNs. ## Requirements - MS or PhD degree in computer science, computer vision, robotics, computer architecture or equivalent experience in technical field (or equivalent experience) - 5+ years of work experience in software development. - 2+ years of experience in **developing** deep learning frameworks (e.g. PyTorch, JAX, TensorFlow, ONNX, etc.) or compiler technologies (e.g. LLVM, MLIR, TVM, Triton, etc.). - Domain experience in technologies used for GPU programming (e.g. CUDA C++ and/or DSLs like OpenAI Triton) or with system-level optimization for deep learning training or inference. - Strong C/C++ programming skills - Familiar with start-of-the-art deep learning techniques for inference and training. - Willing to take action and have strong analytical skills. ## Nice to Have - Experience with MLIR or LLVM or similar compiler technologies - Background with low precision inference, quantization, compression of DNNs - Experience with GPU programming - Experience with building DSLs or optimizing compilers (e.g. graph compiler or kernel generator) for GPUs or other accelerated computing platforms. - Open source project ownership or contribution, healthy GitHub repositories, guiding and/or mentoring experience ## Benefits - Highly competitive salaries - Comprehensive benefits package - Equity - Opportunities for impact on the world through AI technologies - Diverse and supportive work environment - Collaborative and innovative culture NVIDIA offers highly competitive salaries and a comprehensive benefits package. As you plan your future, see what we can offer to you and your family www.nvidiabenefits.com/ Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5. You will also be eligible for equity and benefits. Applications for this job will be accepted at least until January 13, 2026. This posting is for an existing vacancy. NVIDIA uses AI tools in its recruiting processes. NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.