NVIDIA seeks an experienced AI Solutions Architect for a customer-facing role, requiring hands-on expertise in efficient AI model training and deployment. This senior-level position involves collaborating with customers to enhance workload performance and reduce costs, leading technical engagements, developing proof-of-concepts for AI solutions in the Consumer Internet industry, and creating necessary documentation. The ideal candidate will develop and debug software using NVIDIA’s and open-source AI frameworks, working closely with internal teams to secure design wins and drive innovation based on customer feedback. Essential qualifications include a degree in Computer Science or related field, 8+ years of experience coding in Python and C++ with GPUs, proficiency in profiling and optimizing model performance, and strong communication skills for conveying ideas through GitHub, documentation, and presentations. Additional desirable skills encompass full-stack knowledge from deep learning frameworks to lower-level GPU programming, familiarity with MLOps technologies, and expertise in large-scale data pipelines and AI model deployment.
Skills
Python
C++
PyTorch
JAX
CUDA
CUTLASS
cuDNN
NCCL
Kubernetes
MLOps
GitHub
Prometheus
Grafana
CI/CD
What you'll do
Collaborate with customers to enhance workload performance and reduce costs.
Lead proof-of-concept development for AI solutions in the Consumer Internet industry.
Develop and debug software for NVIDIA’s AI frameworks and libraries.
Partner with internal teams to secure design wins based on customer feedback.
Profile and optimize model training/inference performance on GPUs.
Create GPU kernels for deep learning, focusing on GEMM and attention kernels.
What we're looking for
8+ years of experience as an AI/Software Engineer with proven track record in Python/C++.
Experience profiling and optimizing model training/inference performance on GPUs.
Strong background in developing and optimizing GPU kernels for deep learning.
Full stack experience from DL frameworks to lower-level CUDA/CUTLASS/cuDNN/NCCL.
Familiarity with MLOps technologies including containers, Kubernetes, and data center deployments.
Excellent communication skills for conveying technical ideas through documentation and presentations.
Market check
Salary context
This
$184,000–$287,500
range sits above
74%
of similar postings on FindRole.
Peer median band
$169,375–$241,500
Median floor and ceiling across peers.
Typical midpoint (25–75%)
$162,000–$241,750
Middle half of comparable postings.
Based on 240 comparable postings.*
* 240 is the maximum number of comparable postings sampled.
Employer
About Nvidia
Nvidia is a leading designer of graphics processing units (GPUs) and system-on-chip units, powering gaming, professional visualization, data centers, and artificial intelligence workloads. Industry: Semiconductors & AI Computing
Nvidia currently has
801 open roles
on FindRole.
Listed pay typically runs
$184,000–$287,500
across 797 roles with salary data.