Machine Learning Engineer, LLM Post-Training

About this role

NewsBreak is seeking a Machine Learning Engineer to lead the post-training of large language models, primarily focusing on reinforcement learning techniques. This role involves collaborating with product teams to translate requirements into effective training strategies while managing data preparation and evaluation processes.

What you'll do

• Oversee post-training processes involving continuous pre-training, supervised fine-tuning, and reinforcement learning.
• Design and manage data preparation tailored to training needs.
• Collaborate with business and product teams to create training plans.
• Conduct large-scale training on GPU clusters with distributed training techniques.
• Build evaluation and reward pipelines to ensure model quality.
• Stay updated with post-training research and implement new techniques.

What they're looking for

• Hands-on experience with LLM post-training
• Strong data engineering for machine learning
• Large-scale GPU training expertise
• Proficient in PyTorch
• Understanding of tokenization and attention mechanisms
• Fast iteration and business impact orientation
• Communication skills for cross-team collaboration

Benefits

• 100% health, dental, and vision coverage for employees
• Top-tier 401(K) plan with company matching
• Paid time off and holidays
• Flexible spending and health savings accounts
• Team activity budget