Remote Staff ML Researcher: Post-Training & RL

Zekai; United States; Remoto; USD 200.000 - 320.000; Tempo integral

Benefícios oferecidos por esta vaga

Equity compensation

Bonuses

Remote-first workforce

Health insurance reimbursement

401(k) with company match

Resumo da vaga

Zekai is seeking a Member of Technical Staff, Model Training, focusing on post-training research. This is a remote role that blends fundamental ML research with large-scale model development.

You will design and run ambitious experiments, translate research into scalable training methods such as SFT, RL, and distillation, and contribute to model intelligence. The position offers a remote-first environment with strong compensation and growth opportunities.

Qualificações

• Strong foundations in machine learning, deep learning, optimization, and modern model training techniques.
• Hands-on experience training, fine-tuning, or post-training large-scale language or multimodal models.
• Deep technical expertise in one or more areas such as reinforcement learning, model distillation, reward modeling, data selection, or learning dynamics.
• Strong proficiency in Python and PyTorch, with the ability to implement, debug, and optimize complex training pipelines.
• Experience designing controlled experiments, interpreting model behavior, and translating empirical findings into actionable research directions.
• Ability to independently navigate ambiguous research problems, formulate hypotheses, and iterate quickly based on experimental evidence.
• Strong analytical thinking and a research-driven mindset, with an emphasis on reproducibility, scientific rigor, and measurable improvements.

Responsabilidades

• Research, implement, and scale post-training methods across large language and multimodal models (SFT, RL, OPD).
• Investigate learning objectives, optimization dynamics, exploration strategies, and credit assignment to improve training stability and efficiency.
• Develop and refine training recipes by optimizing data mixtures, sampling strategies, curricula, and evaluation methodologies.
• Design methods for learning from expert demonstrations, human preferences, corrections, and model-generated trajectories.
• Research co-evolution of generators, reward models, and graders to improve performance and signaling quality.
• Design rigorous experiments, ablation studies, and evaluation frameworks to measure generalization and reasoning.
• Own research initiatives end-to-end from identifying directions to executing large-scale experiments.

Conhecimentos

Machine learning fundamentals

Deep learning

Optimization

Experiment design

Reinforcement learning

Research mindset

Model training techniques

Ferramentas

Python

PyTorch

Distributed training

Experiment tracking

Descrição da vaga

Zekai is seeking a Member of Technical Staff, Model Training, focusing on post-training research. This is a remote role that blends fundamental ML research with large-scale model development.

You will design and run ambitious experiments, translate research into scalable training methods such as SFT, RL, and distillation, and contribute to model intelligence. The position offers a remote-first environment with strong compensation and growth opportunities.