Senior Deep Learning Algorithm Engineer

NVIDIA is seeking a Senior Deep Learning Algorithm Engineer to join their core AI Frameworks team. You will design, develop, and optimize large language models and multimodal foundation models using open-source frameworks like Megatron Core and NeMo. This role involves innovating on distributed training algorithms, model optimizations, and performance tuning on next-gen GPU architectures.

• Design and implement latest distributed training algorithms
• Optimize large language models and multimodal foundation models
• Contribute to open-source NeMo-RL, Megatron Core, and NeMo Framework
• Solve large-scale AI training and inference challenges
• Performance tuning and optimizations on NVIDIA GPU architectures
• Research, prototype, and develop robust AI tools and pipelines