Enterprise AI Solutions Architect: Scale Generative Models

NVIDIA Corporation; United States; Remote; USD 152,000 - 242,000; Full time

Benefits offered by this job

Equity

Benefits

Job summary

NVIDIA Corporation is seeking a Solutions Architect to join the AI Software Segment SA team, shaping enterprise AI solutions with NVIDIA's latest technology. You will collaborate with customers and partners to solve complex problems at scale across data processing, training, and deployment.

The role requires strong experience with AI frameworks (PyTorch, JAX, TensorFlow), Python, Linux, and distributed computing, plus the ability to work with executives and engineering teams.

Qualifications

• BS, MS, or PhD in Engineering, Mathematics, Physics, Computer Science, Data Science, or equivalent.
• 5+ years of experience with AI frameworks (PyTorch, JAX, TensorFlow) and libraries like Hugging Face Transformers.
• Proficiency in Python programming, Linux environment, with 5+ years of experience.
• Hands-on experience with AI model lifecycle (evaluation, pre-training, post-training, RL, optimization).
• Expertise in distributed computing, model and data parallelism.
• Experience with SLURM and Kubernetes for training large models on GPUs.
• Clear written and oral communication skills with ability to collaborate with executives and engineering teams.

Responsibilities

• Develop end-to-end AI solutions for enterprise use cases and help customers adopt NVIDIA AI models.
• Apply the full AI model lifecycle from data processing to training, RL, evaluation, and optimization.
• Support a broad model portfolio including LLMs, multimodal, retrieval, speech, and edge use cases.
• Partner with enterprise customers in co-design engagements to meet data, evaluation criteria, and success metrics.
• Improve NVIDIA products and build creative solutions to scaling challenges across architecture, models, libraries, and AI applications.
• Share expertise by contributing to open-source NVIDIA libraries and delivering hands-on training.

Skills

Python programming

Linux

Distributed computing

Communication skills

Learning agility

Problem solving in AI workloads

Education

Bachelor's/Master's/PhD in Engineering/CS/Math

Tools

PyTorch

JAX

TensorFlow

Hugging Face Transformers

SLURM

Kubernetes

Job description

NVIDIA Corporation is seeking a Solutions Architect to join the AI Software Segment SA team, shaping enterprise AI solutions with NVIDIA's latest technology. You will collaborate with customers and partners to solve complex problems at scale across data processing, training, and deployment.

The role requires strong experience with AI frameworks (PyTorch, JAX, TensorFlow), Python, Linux, and distributed computing, plus the ability to work with executives and engineering teams.