AI Performance & Kernel Engineer for Frontier-Scale ML
Zyphra
14 days+
Get more replies from employers
Send a job-specific resume in minutes.
Benefits offered by this job
Job summary
A leading AI technology firm located in San Francisco is seeking a Research Engineer specializing in AI Performance & Kernel Optimization. The role involves enhancing the performance of large-scale AI systems, optimizing kernels, and collaborating with various teams. Ideal candidates should have a strong engineering background in GPU kernel development and experience with ML workloads. Benefits include comprehensive health plans, competitive compensation, and an engaging work environment.
Qualifications
- Experience writing highly performant GPU kernels.
- Experience optimizing ML workloads for large-scale training.
- Strong understanding of distributed training systems.
Responsibilities
- Improve performance of language model training and inference stacks.
- Optimize kernel development for ML workloads.
- Collaborate with research and infrastructure teams.
Skills
Building reliable, high-performance systems
Low-level performance intuition
Excellent communication skills
Collaboration across teams
Performance tuning expertise
Education
Background in physics, mathematics, computer science, or electrical engineering
Tools
PTX
CUDA
HIP
Triton
Job description
A leading AI technology firm located in San Francisco is seeking a Research Engineer specializing in AI Performance & Kernel Optimization. The role involves enhancing the performance of large-scale AI systems, optimizing kernels, and collaborating with various teams. Ideal candidates should have a strong engineering background in GPU kernel development and experience with ML workloads. Benefits include comprehensive health plans, competitive compensation, and an engaging work environment.
'/%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}
