ML Engineer — LLM Post-Training & RL Specialist

NewsBreak; Mountain View (CA); On-site; USD 130,000 - 160,000; Full time

Benefits offered by this job

Health, dental, and vision care

401(k) plan with company matching

Paid time off and holidays

Job summary

NewsBreak in Mountain View, California, is seeking a Machine Learning Engineer to lead the post-training of large language models. You will drive continuous pre-training, supervised fine-tuning, and reinforcement learning while collaborating closely with product and business teams.

The ideal candidate has hands-on experience with RL techniques and strong data engineering skills, particularly with GPU training. This role offers competitive benefits such as health coverage and a top-tier 401(k) plan.

Qualifications

• Hands-on experience with continuous pre-training, supervised fine-tuning, and reinforcement learning.
• Ability to design data preparation plans for business scenarios.
• Experience with distributed training on mid-to-large GPU hardware.

Responsibilities

• Lead post-training of LLMs across the full pipeline.
• Design and build data that drives each training stage.
• Partner with stakeholders to understand business requirements.

Skills

Hands-on LLM post-training experience

Strong data engineering for ML

Proven large-scale GPU training ability

Strong PyTorch fundamentals

Understanding of tokenization and attention

Fast iteration and business impact

Tools

PyTorch

Hugging Face TRL/Accelerate

DeepSpeed

Job description

NewsBreak in Mountain View, California, is seeking a Machine Learning Engineer to lead the post-training of large language models. You will drive continuous pre-training, supervised fine-tuning, and reinforcement learning while collaborating closely with product and business teams.

The ideal candidate has hands-on experience with RL techniques and strong data engineering skills, particularly with GPU training. This role offers competitive benefits such as health coverage and a top-tier 401(k) plan.