Role overview
NVIDIA is looking for a dedicated and motivated senior build and continuous integration (CI/CD) engineer for its GenAI Frameworks ( Megatron-LM and NeMo Framework ) team. Megatron-LM and NeMo Framework are open-source, scalable and cloud-native frameworks built for researchers and developers working on Large Language Models (LLM), Multimodal (MM), and Video Generation. Megatron-LM and NeMo Framework provide end-to-end model training, including data curation, alignment, customization, evaluation, deployment and tooling to optimize performance and user experience. Building upon the latest DevOps tools, your work will enable GenAI framework software engineers, deep learning algorithm engineers, and research scientists to work efficiently with a wide variety of deep learning algorithms and software stacks as they vigilantly seek out opportunities for performance optimization and continuously deliver high quality software.
Does the idea of pushing the boundaries of innovative research and development excite you? Are you interested in getting exposure to the entire DL SW stack? Then join our technically diverse team of DL algorithm engineers and performance optimization specialists to unlock unprecedented deep learning performance in every domain.
What we're looking for
- BS or MS degree in Computer Science, Computer Architecture or related technical field (or equivalent experience) and 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 and experience with solutions like Gitlab, Github, Jenkins etc.
- Experience with Linux system administration.
- Proficient with containerization and cluster management technologies like Docker and Kubernetes.
- Experience in build tools, including Make, Cmake.
- A strong background in source code management (SCM) solutions such as GitLab, GitHub, Perforce, etc.
- Well-versed problem-solving and debugging skills.
- Great teammate who can collaborate and influence others in a dynamic environment.
- Excellent interpersonal and written communication skills.
- Proven-track record with GPU accelerated systems at scale.
- Well-versed in DL frameworks such as PyTorch, Jax, or TensorFlow.
- Expertise in cluster and cloud compute technologies, e.g.: SLURM, Lustre, k8s
- Software and hardware Benchmarking on high-performance computing systems.
Your base salary will be determined based on your location, experience, and the pay of employees in similar positions. The base salary range is 184,000 USD - 287,500 USD for Level 4, and 224,000 USD - 356,500 USD for Level 5.
You will also be eligible for equity and benefits .
Applications for this job will be accepted at least until August 18, 2025.NVIDIA is committed to fostering a diverse work environment and proud to be an equal opportunity employer. As we highly value diversity in our current and future employees, we do not discriminate (including in our hiring and promotion practices) on the basis of race, religion, color, national origin, gender, gender expression, sexual orientation, age, marital status, veteran status, disability status or any other characteristic protected by law.
JR2000137