Solutions Architect, AI Models jobs in United States
cer-icon
Apply on Employer Site
company-logo

NVIDIA · 21 hours ago

Solutions Architect, AI Models

NVIDIA is the world leader in GPU accelerated computing and AI, seeking a Solutions Architect to join the AI Enterprise Segment team. The role involves developing end-to-end AI solutions for enterprise use cases, guiding customers in implementing advanced AI models, and contributing to the community through knowledge sharing and open-source projects.

AI InfrastructureArtificial Intelligence (AI)Consumer ElectronicsFoundational AIGPUHardwareSoftwareVirtual Reality
check
Growth Opportunities
check
H1B Sponsor Likelynote

Responsibilities

A huge part of our work involves developing end-to-end AI solutions for enterprise use cases. We help customers adopt NVIDIA AI SDKs and APIs by offering deep technical expertise
Tackle sophisticated AI challenges by applying skills across the AI model lifecycle—from data preprocessing and orchestration to training, post-training, evaluation, and optimized deployment
Guide customers in implementing next-generation model distillation, domain adaption, reinforcement learning (RL) and post-training algorithms, using NVIDIA frameworks
As we work with customers across multiple industries, we help improve NVIDIA products and build creative solutions to overcome scaling challenges at the intersection of computer architecture, libraries, and AI applications
Contribute to the wider organization and community by sharing your expert knowledge. This can vary from contributing to open-source projects and product engineering to publishing findings and delivering hands-on training

Qualification

AI frameworksPython programmingAI model lifecycleDistributed computingLinux environmentClear communicationAdaptability

Required

Strong foundational expertise, from a BS, MS, or Ph.D. degree in Engineering, Mathematics, Physics, Computer Science, Data Science, or similar (or equivalent experience)
5+ years of experience with AI frameworks such as PyTorch, JAX, or TensorFlow, and libraries like Hugging Face Transformers
Proficiency in Python programming, software design, debugging, and performance analysis, with at least 5+ years of experience in a Linux environment
Hands-on experience with full AI model lifecycle, including pre-training, supervised finetuning, post-training techniques such as reinforcement learning (RL), and model evaluation
Expertise in distributed computing methodologies, including model and data parallelism
Experience with distributed computing tools, like SLURM and Kubernetes, for training large models on GPUs
Ability to learn fast and quickly adapt to change
Clear written and oral communications skills with the ability to effectively collaborate with executives and engineering teams

Preferred

Experience with and/or contributions to open-source NVIDIA AI Enterprise deep learning libraries and frameworks, particularly NeMo, Megatron Core, or NeMo-RL
Hands-on experience in large-scale foundation model training, accuracy, and performance profiling
Prior experience with AI model training techniques applied to multi-modal data (audio, image, and video)
Knowledge of NVIDIA GPU/CPU architecture and its impact on software performance
Show willingness and ability to dig into unfamiliar territories to solve complex problems relying on experience from previous work

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