Tech Lead Manager (AI)

TL;DR

Tech Lead Manager (AI): Building and optimizing a distributed framework for large language model training with an accent on system optimization and collaboration with ML teams. Focus on integrating state-of-the-art technologies and enhancing ML research capabilities.

Location: San Francisco, CA; New York, NY

Salary: $252,000—$315,000 USD

Company

Scale develops reliable AI systems for critical decisions, providing high-quality data and technologies for leading models.

What you will do

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

Requirements

• Experience with multi-node LLM training and inference.
• Experience with developing large-scale distributed ML systems.
• Strong software engineering skills, proficient in frameworks such as CUDA and Pytorch.
• Strong written and verbal communication skills.

Nice to have

• Expertise in post-training methods and next-generation use cases for large language models.