GrepJob
NewsBreak

Machine Learning Engineer, LLM Post-Training

NewsBreak
Apply
about 18 hours ago
Mountain View, CA, USAMid Level / Senior
H1B Sponsor

Base Salary

$150k - $230k/yr

Responsibilities

  • Lead post-training of LLMs across the full pipeline: continuous pre-training, SFT, and reinforcement learning.
  • Design, build, and curate data for each training stage tailored to specific business needs.
  • Collaborate with business and product stakeholders to convert requirements into training plans.
  • Run large-scale training on mid-to-large GPU clusters using distributed-training techniques.
  • Build and maintain evaluation and reward pipelines to measure model quality.
  • Stay current with post-training research and implement promising techniques.

Requirements

  • Hands-on LLM post-training experience with practical RL experience.
  • Strong data engineering skills for ML, including data sourcing and preparation.
  • Proven ability to train LLMs on mid-to-large GPU hardware.
  • Strong fundamentals in PyTorch and familiarity with relevant frameworks.
  • Solid understanding of tokenization, attention, and common failure modes in training.
  • A bias toward fast iteration and business impact with strong communication skills.

Benefits

  • Health, dental, and vision care for you and your family with 100% coverage for employees.
  • Top-tier 401(K) plan with company matching.
  • Paid time off and paid holidays.
  • FSA, HSA, and commuter benefits programs.
  • Team activity budget.

Tech Stack

Categories

AI & MLData Engineering