NVIDIA · 2 days ago
Senior Software Engineer, Deep Learning - MLIR TRT
Santa Clara, CA
Full-time
Onsite
Senior Level
$184K/yr - $288K/yr
5+ years expNVIDIA is a leading technology company specializing in computer graphics and AI solutions. They are seeking a Senior Deep Learning Software Engineer to develop and productize deep learning solutions for autonomous driving vehicles, focusing on compiler technologies and performance optimization for NVIDIA's hardware platforms.
AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
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
Qualification
Required
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
Preferred
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
Equity
Benefits
Company
NVIDIA
NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.
10001+ employees
H1B Sponsorship
NVIDIA has a track record of offering H1B sponsorships. Please note that this does not
guarantee sponsorship for this specific role. Below presents additional info for your
reference. (Data Powered by US Department of Labor)
Distribution of Different Job Fields Receiving Sponsorship
Represents job field similar to this job
Trends of Total Sponsorships
2025 (1877)
2024 (1355)
2023 (976)
2022 (835)
2021 (601)
2020 (529)
Funding
Current Stage
Public CompanyTotal Funding
$4.09BKey Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity
Leadership Team
Jensen Huang
Founder and CEO
Michael Kagan
Chief Technology Officer
Recent News
2026-01-25
Unified Communications fuel big enterprise success | CIO
2026-01-25
2026-01-25
Company data provided by crunchbase