Senior Deep Learning Performance Architect jobs in United States
cer-icon
Apply on Employer Site
company-logo

NVIDIA · 5 days ago

Senior Deep Learning Performance Architect

NVIDIA is looking for outstanding Performance Architects with a background in performance analysis, performance modeling, and AI/deep learning. The role involves developing innovative architectures to enhance deep learning performance and collaborating with various teams to guide the direction of deep learning hardware and software.

AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
check
Growth Opportunities
check
H1B Sponsor Likelynote
Hiring Manager
Joshua Hasten
linkedin

Responsibilities

Develop innovative architectures to extend the state of the art in deep learning performance and efficiency
Analyze performance, cost and power trade-offs by developing analytical models, simulators and test suites
Understand and analyze the interplay of hardware and software architectures on future algorithms, programming models and applications
Evaluate PPA (performance, power, area) for hardware features and system level architectural trade-offs. Develop high level simulators in C++/Python
Actively collaborate with software, product and research teams to guide the direction of deep learning HW and SW

Qualification

Deep Learning ArchitecturePerformance ModelingGPU ArchitecturePython ProgrammingC++ ProgrammingMachine LearningCritical ThinkingSelf-Motivation

Required

MS or PhD in Computer Science, Computer Engineering, Electrical Engineering or equivalent experience
6+ years of relevant meaningful work experience
Strong background in GPU or Deep Learning ASIC architecture for distributed training and/or inference spanning multi-chip/multi-node
Experience with performance modeling, architecture simulation, profiling, and analysis
Solid foundation in machine learning and deep learning. Understanding of modern transformer-based architectures and their performance at scale
Strong programming skills in Python, C, C++

Preferred

Background with deep neural network training, inference and optimization in leading frameworks (e.g. Pytorch, JAX, TensorRT)
Familiarity with advanced optimizations and SW/HW co-design in LLM training and inference
Exposure to using AI to accelerate SW engineering
Demonstration of self-motivation and creative / critical thinking

Benefits

Equity
Benefits

Company

NVIDIA is a computing platform company operating at the intersection of graphics, HPC, and AI.

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 Company
Total Funding
$4.09B
Key Investors
ARPA-EARK Investment ManagementSoftBank Vision Fund
2023-05-09Grant· $5M
2022-08-09Post Ipo Equity· $65M
2021-02-18Post Ipo Equity

Leadership Team

leader-logo
Jensen Huang
Founder and CEO
linkedin
leader-logo
Michael Kagan
Chief Technology Officer
linkedin
Company data provided by crunchbase