Senior GenAI Research Engineer - Optimization and Kernels

We are obsessed with enabling data teams to solve the world's toughest problems. As a research engineer on the Scaling team, you will be responsible for keeping up with the latest developments in deep learning and advancing the scientific frontier by creating new techniques that go beyond the state of the art.

Requirements

• BS/MS/PhD in Computer Science or related field
• Hands-on experience writing and tuning CUDA kernels for ML training applications
• Strong understanding of NVIDIA GPU architecture and proficiency with CUDA debugging/profiling tools
• Deep understanding of parallelism techniques and memory optimization strategies for large-scale model training
• Strong software engineering skills in Python and PyTorch

Benefits

• Comprehensive benefits and perks
• Annual performance bonus
• Equity
• Paid Time Off