Tech Lead Manager for Scalable LLM Training Platform
Scale AI
14 days+
Application generator
Get a reply from this employer — a resume and cover letter tailored to exactly what they’re hiring for.
Get past ATS filters
Job summary
A technology company specializing in AI solutions in San Francisco seeks an experienced software engineer. You will build and optimize large-scale ML frameworks, collaborating with cross-functional teams and integrating state-of-the-art technologies. The ideal candidate is passionate about system optimization and has experience with multi-node LLM training. This role offers a chance to influence the future of AI innovation.
Qualifications
- Passionate about system optimization.
- Experience with multi-node LLM training and inference.
- Strong written and verbal communication skills.
Responsibilities
- Build, profile, and optimize our training and inference framework.
- Collaborate with ML and research teams.
- Research and integrate state-of-the-art technologies.
Skills
System optimization
Multi-node LLM training
Large-scale distributed ML systems
Post-training methods (RLHF/RLVR)
Software engineering (CUDA, Pytorch)
Communication skills
Tools
CUDA
Pytorch
Transformers
Flash attention
Job description
A technology company specializing in AI solutions in San Francisco seeks an experienced software engineer. You will build and optimize large-scale ML frameworks, collaborating with cross-functional teams and integrating state-of-the-art technologies. The ideal candidate is passionate about system optimization and has experience with multi-node LLM training. This role offers a chance to influence the future of AI innovation.
'/%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}
