Senior MLOps Engineer, GenAI Framework

Job Description

NVIDIA is seeking a senior build and continuous integration engineer for its GenAI Frameworks team to work on Megatron-LM and NeMo Framework, open-source frameworks for Large Language Models, Multimodal, and Video Generation.

Requirements

• BS or MS degree in Computer Science, Computer Architecture or related technical field (or equivalent experience)
• 6+ years of industry experience in DevOps and infrastructure engineering
• Strong system level programming in languages like Python and shell scripting
• Extensive understanding of build/release systems, CI/CD using tools like Gitlab, Github, Jenkins
• Experience with Linux system administration
• Proficient with containerization and cluster management technologies like Docker and Kubernetes
• Experience with build tools such as Make, Cmake
• Strong background in source code management solutions like GitLab, GitHub, Perforce
• Problem-solving and debugging skills
• Good collaboration and communication skills

Responsibilities

• Architect and manage continuous integration pipelines and release processes for Generative AI frameworks related to Megatron-LM and NeMo Framework
• Design and implement efficient and scalable DevOps solutions
• Work with industry standard tools in hybrid on-premise and cloud environments
• Assist with cluster operations and system administration
• Automate recurring tasks for accelerating research and development cycles
• Develop new quality control measures and best practices
• Collaborate with DL frameworks and libraries teams and other engineering teams within NVIDIA

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