Join NVIDIA and help build the software that will define the future of generative AI. We are looking for a research engineer who is passionate about open-source and excited to create our next-generation post-training software stack.
Requirements
- BS, MS or PhD in Computer Science, AI, Applied Math, or related fields or equivalent experience
- 3+ years of proven experience in machine learning, systems, distributed computing, or large-scale model training.
- Experience with AI Frameworks such as Pytorch or JAX
- Experience with at least one inference and deployment environments such as vLLM, SGLang or TRT-LLM
- Proficient in Python programming, software design, debugging, performance analysis, test design and documentation.
- Strong understanding of AI/Deep-Learning fundamentals and their practical applications.
