Senior GenAI Kernel & GPU Optimization Engineer
Databricks
Get more replies from employers
Send a job-specific resume in minutes.
Benefits offered by this job
Job summary
Databricks is looking for a Sr. Research Engineer to join their Scaling Team in Mountain View, California. The role involves driving performance improvements and designing high-performance GPU kernels for training workloads. Candidates should have strong experience in CUDA programming, distributed training frameworks, and a solid understanding of NVIDIA GPU architecture.
The position offers a competitive salary range of $166,000 to $225,000 USD, alongside comprehensive benefits. Join us to advance AI models and deliver cutting-edge solutions to make our customers successful.
Qualifications
- Hands-on experience writing and tuning CUDA kernels for ML training.
- Experience in distributed training frameworks like PyTorch DDP or DeepSpeed.
- Strong understanding of memory optimization strategies for large-scale model training.
Responsibilities
- Drive performance improvements through advanced optimization techniques.
- Design and implement distributed training frameworks for large language models.
- Profile, debug, and optimize end-to-end training workflows.
Skills
Education
Tools
Job description
Databricks is looking for a Sr. Research Engineer to join their Scaling Team in Mountain View, California. The role involves driving performance improvements and designing high-performance GPU kernels for training workloads. Candidates should have strong experience in CUDA programming, distributed training frameworks, and a solid understanding of NVIDIA GPU architecture.
The position offers a competitive salary range of $166,000 to $225,000 USD, alongside comprehensive benefits. Join us to advance AI models and deliver cutting-edge solutions to make our customers successful.
'/%3e%3cpath%20d='M101.744%2012.4726L102.894%2016.1947L99.0021%2013.2201L101.744%2012.4726ZM105.3%208.40937H100.488L99.0026%203.59473L97.5116%208.41047L92.7%208.40385L96.5965%2011.3834L95.1055%2016.1942L99.0021%2013.2196L101.408%2011.3834L105.3%208.40992V8.40937Z'%20fill='white'/%3e%3cdefs%3e%3clinearGradient%20id='paint0_linear_5593_1381'%20x1='90'%20y1='9.44471'%20x2='99'%20y2='9.44471'%20gradientUnits='userSpaceOnUse'%3e%3cstop%20stop-color='%2300AD70'/%3e%3cstop%20offset='0.998594'%20stop-color='%2300B67A'/%3e%3cstop%20offset='1'%20stop-color='%23DCDCE6'/%3e%3c/linearGradient%3e%3c/defs%3e%3c/svg%3e){kind=small}
