Tech Lead Manager for Scalable LLM Training Platform

Scale AI; San Francisco (CA); On-site; USD 120,000 - 160,000; Full time

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.