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. The role involves collaborating with customers, leading customer technical engagements, and developing 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.
- 8+ 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.
Benefits
- Generous Paid Time Off
- 401k Matching
- Retirement Plan
- Equity