Principal ML Researcher
AI
Toloka AI is seeking a Principal ML Researcher to define the technical vision and architecture for its core ML and post-training stack. The role involves leading research on LLM alignment, RL frameworks, evaluation systems, and model optimization, while mentoring senior engineers and collaborating with executive leadership. Requires 6+ years in ML engineering or applied research, with deep expertise in LLM post-training and distributed training systems.
What you'll do
- Drive technical strategy for Toloka’s post-training stack
- Architect greenfield post-training and RL pipelines
- Pioneer next-gen evaluation harnesses
What you need
Must-have
PythonPyTorchDeepspeedMegatronFSDPvLLMTensorRT-LLMSFT
Soft skills
Technical leadershipMentorshipCommunicationCross-functional collaborationArchitectural visionProduct leadership
Machine LearningLLMPost-trainingReinforcement LearningEvaluationRemoteAI Research