Solutions Architect, Agentic Optimization

NVIDIA is looking for an AI Solutions Architect with hands-on experience in efficient AI model training and/or deployment for a customer-facing role.

Primary responsibilities will be to help accelerate customer workloads and lead customer technical engagements around NVIDIA software and technologies.

Responsibilities:

• Collaborate closely with customers to improve their workload performance and reduce infrastructure costs.
• Lead and develop proof-of-concepts for AI solutions applied to the Consumer Internet industry, including areas like LLMs and recommenders, and build collateral (notebook/code) as needed.
• Develop and debug software for NVIDIA and open-source AI frameworks and libraries.
• Partner with NVIDIA's software engineering, product, and sales teams to secure design wins and drive the development of innovative solutions based on customer feedback.

Requirements:

• BS, MS, or PhD in Computer Science, Electrical/Computer Engineering, Physics, Mathematics, or other Engineering fields, or equivalent experience.
• 5+ years of experience as an AI/Software Engineer with a proven track record coding in Python and/or C++ with popular AI software libraries and GPUs.
• Experience with profiling and optimizing model training/inference performance on GPUs.
• Experience developing and optimizing GPU kernels for deep learning, with a focus on GEMM and attention kernels.
• Strong communication skills, with the ability to clearly convey ideas and code through GitHub, documentation, and presentations.
• A great teammate who enjoys collaborating with cross-functional teams including Engineering, Research, Sales, Product, and Marketing.
• Self-starter with a passion for growth, continuous learning, and sharing insights with the team.

Nice to Have:

• Full stack experience.
• Experience working with enterprise developers and strong customer-facing skills.
• Familiarity with MLOps technologies such as containers, Kubernetes, and data center deployments.
• Experience with large-scale production data pipelines and AI model training/deployment.
• Creative problem-solving skills for debugging and resolving complex issues.

You will also be eligible for equity and benefits.