AI Research Scientist

About the role

NVIDIA is looking for AI researchers to join its Research Singapore team, which develops and optimizes GPU-accelerated efficient AI computing across language models, visual generation and robotics. The team focuses on making generative AI models both powerful and computationally efficient, collaborates with others and regularly publishes at top venues in machine learning, robotics and computer vision.

Key responsibilities

• Research, design and accelerate novel generative AI models on NVIDIA GPUs
• Publish original research
• Collaborate with team members and external researchers
• Mentor interns
• Speak at conferences and events
• Transfer technology to product groups

Required profile

• PhD in computer science or electrical engineering with a strong research track record
• 4 years of experience
• Excellent knowledge of LLMs and foundation models
• Publication at leading conferences such as CVPR, ICML, ICLR, ICCV, NeurIPS or MLSys
• Excellent communication skills

Required skills

• Rapid prototyping languages such as Python
• C++ and parallel programming (e.g. CUDA) is a plus
• Machine learning frameworks such as PyTorch