NVIDIA is seeking an AI Solutions Architect to enhance customer workloads and lead technical engagements involving NVIDIA software and technologies. This role focuses on collaborating with customers to optimize performance and reduce costs while developing proof-of-concepts for AI solutions. The position requires hands-on experience in AI model training and deployment, particularly within the Consumer Internet industry. The successful candidate will work closely with various teams to drive innovative solutions based on customer feedback.

• 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 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.

• 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.
• 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.

• Employees at NVIDIA are often offered comprehensive, day-one benefits—including medical, dental, and vision coverage with HSA support, life and disability insurance, an Employee Assistance Program, and a 401(k) with auto-enrollment. Many roles also have generous time off and holidays, donation matching (up to $10,000), and a wide menu of extras like FSAs, commuter benefits, legal and identity-theft protection, pet insurance, and wellness discounts. Optional programs can include student-loan and home-purchase support, plus family care resources and expert medical services.