Solutions Architect, AI Models

What you'll do

• Develop end-to-end AI solutions for enterprise use cases using NVIDIA AI models and libraries.
• Apply technical expertise across the AI model lifecycle including data processing, training, and optimization.
• Support a broad portfolio of models including LLMs, multimodal, speech, and edge use cases.
• Partner with enterprise customers in co-design engagements to deliver customized AI solutions based on specific success metrics.
• Solve scaling challenges at the intersection of computer architecture, models, libraries, and AI applications.
• Contribute to the wider community by contributing to open-source projects and delivering hands-on training.
• Act as a trusted expert advisor to customers and partners to solve hard problems in deploying AI workloads at scale.

What we're looking for

• BS, MS, or Ph.D. degree in Engineering, Mathematics, Physics, Computer Science, Data Science, 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 5+ years of experience in a Linux environment.
• Hands-on experience with the AI model lifecycle, including evaluation, failure analysis, pre-training, post-training, RL, and model optimization.
• Expertise in distributed computing methodologies, including model and data parallelism.
• Experience with distributed computing tools like SLURM and Kubernetes for training and evaluating large models on GPUs.
• Clear written and oral communication skills to collaborate with executives and engineering teams.
• Experience with open-source NVIDIA AI libraries, data curation for RL, multi-modal training, or GPU architecture knowledge (preferred).