Machine Learning Software Platform Architect

Job Description

NVIDIA is seeking a highly skilled and experienced Large Language Model (LLM) based Application Infrastructure engineer to join their team at the intersection of GPU chip design and AI.

Requirements

• BS in computer science or related field or equivalent experience
• 5+ years of experience in developing and maintaining AI or machine learning infrastructure, preferably with large language models
• Strong proficiency in Python and web development
• Understanding of chip design and related computational and data challenges
• Experience with data management, including doc cleaning, transformation, and secure storage
• Excellent problem-solving skills and ability to work effectively in a team
• In-depth understanding of Machine Learning / Deep Learning / NLP concepts

Responsibilities

• Develop and maintain infrastructure for managing large language models (LLMs) based applications for chip design and hardware domain
• Collaborate with HW chip designers and LLM research teams to understand specific needs and challenges of GPU design
• Optimize infrastructure for performance, scalability, and reliability
• Stay updated with latest industry trends in AI and machine learning

Benefits

• Multiple relocation packages
• Two weeklong shutdowns (mid-summer and year-end) in the US (in addition to PTO)
• 8-week parental leave
• 9 Employee Resource Groups
• Annual bonus offering
• Flexible work arrangements
• Up to 6% 401K matching