Senior GenAI Research Engineer (AI)

TL;DR

Senior GenAI Research Engineer (AI): Developing advanced optimization techniques and high-performance GPU kernels for large language models with an accent on distributed training frameworks and performance improvements. Focus on designing and implementing cutting-edge solutions for training workloads and optimizing end-to-end workflows.

Location: San Francisco, California

Salary: $166,000—$225,000 USD

Company

Databricks is the data and AI company, providing a platform for data intelligence used by over 10,000 organizations worldwide.

What you will do

• Drive performance improvements through advanced optimization techniques.
• Design, implement, and optimize high-performance GPU kernels for training workloads.
• Implement distributed training frameworks for large language models.
• Profile, debug, and optimize end-to-end training workflows.

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.
• Proficiency with CUDA debugging/profiling tools.
• Strong software engineering skills in Python and PyTorch.