Tech Lead Manager for Scalable LLM Training Platform

Scale AI; San Francisco (CA); Presencial; USD 120.000 - 160.000; Jornada completa

Descripción de la vacante

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.

Formación

• Passionate about system optimization.
• Experience with multi-node LLM training and inference.
• Strong written and verbal communication skills.

Responsabilidades

• Build, profile, and optimize our training and inference framework.
• Collaborate with ML and research teams.
• Research and integrate state-of-the-art technologies.

Conocimientos

System optimization

Multi-node LLM training

Large-scale distributed ML systems

Post-training methods (RLHF/RLVR)

Software engineering (CUDA, Pytorch)

Communication skills

Herramientas

CUDA

Pytorch

Transformers

Flash attention

Descripción del empleo

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.