Tech Lead Manager- MLRE, ML Systems
Core
Building and optimizing the internal distributed framework for large language model (LLM) training, inference, and data curation to support MLEs, researchers, and data scientists.
Role type
Tech Lead Manager for ML Systems (LLM Post-Training)
Builds
Distributed training and inference framework for next-generation LLMs
Domain
Artificial Intelligence / Large Language Models / Distributed Systems
Deliverable
production ML models | infrastructure
Required skills
System optimization, multi-node LLM training and inference, large-scale distributed ML systems, post-training methods (RLHF/RLVR/PPO/GRPO), software engineering, CUDA, PyTorch, transformers, flash attention
Preferred skills
Expertise in post-training methods, next-generation LLM use cases (instruction tuning, tool use, reasoning, agents, multimodal)
Technologies
CUDA, PyTorch, transformers, flash attention
Responsibilities
Build, profile, and optimize training and inference frameworks; Collaborate with ML and research teams to accelerate R&D; Research and integrate state-of-the-art technologies to optimize ML systems
Seniority
Senior, hands-on IC with leadership responsibilities