# AI Solutions Engineer

**Company**: NVIDIA
**Experience**: senior
**Job type**: full-time
**Category**: Engineering
**Industry**: Technology

**Apply**: https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/China-Shanghai/NIM-Solution-Architect_JR2025729?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply
**Canonical**: https://yubhub.co/jobs/job_60d8b6a1-c4c

## Description

We are seeking an AI Solutions Engineer to join our team, focusing on NVIDIA Inference Microservices (NIM), inference and reinforcement learning (RL) rollout performance, and AI workflow enablement for large language models (LLM), vision language models (VLM), and other generative AI workloads.

**Responsibilities:**

- Drive the implementation, deployment, and optimization of NVIDIA Inference Microservices (NIM) solutions for enterprise and industry AI workloads.

- Package and serve open-source, NVIDIA, and customer-proprietary models through NIM with standardized, containerized APIs for on-premises, cloud, and hybrid environments.

- Optimize high-volume inference and rollout workloads for LLMs and VLMs.

- Evaluate and tune NIM models.

- Deliver technical projects, demos, and client support tasks as directed by the Solution Architecture Leadership.

- Provide technical support and guidance to customers, facilitating the adoption and implementation of NVIDIA technologies and products.

- Collaborate with cross-functional teams to enhance and expand our AI solutions portfolio.

**Requirements:**

- Master's degree or higher in Computer Science, Machine Learning, Electrical Engineering, Mathematics, or a related technical field, or equivalent experience.

- 2+ years of hands-on experience in machine learning engineering, applied research, LLM/VLM inference, or RL rollout.

- Production-quality Python and PyTorch skills, including distributed GPU training, solution, profiling, debugging, memory optimization.

- Working knowledge of transformer architectures, performance optimization, rollout sampling strategies, structured generation, and model-quality evaluation.

- Strong written and verbal communication skills, with the ability to collaborate effectively across research, engineering, infrastructure, product, and customer-facing teams.

**Nice to Have:**

- Publications, open-source contributions, or significant technical projects, LLM/VLM, agent systems.

- Experience applying programmatic verification, simulators, compilers, execution sandboxes, APIs, or external tools as reward sources for model training.

- Familiarity with open-source RL frameworks such as SLIME, Nemo-RL.

- Familiarity with enterprise AI deployment, customer adaptation, or adapting foundation models to specialized vertical domains.

## Skills

### Required
- Python
- PyTorch
- Machine Learning
- LLM
- VLM
- Reinforcement Learning
- Transformer Architectures

### Nice to have
- SLIME
- Nemo-RL
- Programmatic Verification
- Simulators
- Compilers
- Execution Sandboxes
- APIs

---

Source: [Apply at nvidia.wd5.myworkdayjobs.com](https://nvidia.wd5.myworkdayjobs.com/en-US/NVIDIAExternalCareerSite/job/China-Shanghai/NIM-Solution-Architect_JR2025729?utm_source=yubhub.co&utm_medium=jobs_feed&utm_campaign=apply)