N

NextToken

Post-training systems for continuous AI improvement

Member of Technical Staff - Research Engineering

AI ResearchConfirmed live in the last 24 hours
Full-timeNo salary listedIn-person

About the Job

Requirements

  • Understanding of modern post-training methods, including reinforcement learning with verifiable rewards, policy optimization approaches such as GRPO, and on-policy distillation and self-distillation; practical depth in at least one relevant area.
  • Familiarity with how frameworks such as PyTorch or JAX execute model training, and ability to inspect research code, diagnose training problems, and assess whether an implementation matches the intended method.
  • Experience investigating model training, evaluations, agent environments, or related machine learning research, with ability to explain personal contribution and supporting evidence.
  • Strong experimental judgment, including well-chosen baselines, controlled comparisons, and attention to generalization, data leakage, and misleading rewards or metrics.
  • Ability to turn an open-ended problem into a research plan, work through unfamiliar systems, and explain technical decisions to teammates and customers.

Responsibilities

  • Develop post-training methods for specialized language models and agents, including designing and running training recipes that combine supervised fine-tuning, reinforcement learning, and distillation, and investigating effects on capability, generalization, and efficiency.
  • Research synthetic data generation and curation, exploring teacher selection, trajectory sampling, and data mixtures that produce useful learning signals from production tasks and feedback.
  • Develop agent environments, reward models, and verifiers for training and evaluation; establish benchmarks that capture meaningful product improvements and expose weaknesses in model behavior.
  • Work with systems engineers to improve experiment and model-inference efficiency, connecting algorithmic choices to rollout cost, memory use, and training stability.
  • Carry promising research into customer applications and turn successful approaches into reusable methods and tools; use production behavior to identify subsequent research questions.
  • Communicate findings through technical writing, research publications, and open-source contributions where appropriate.

Desired Qualifications

  • Experience with agentic reinforcement learning, reward modeling, synthetic data, distillation, or efficient inference is especially relevant.
  • Research papers, open-source contributions, and independently built systems are useful ways to demonstrate relevant work.

About the Company

NextToken

Post-training systems for continuous AI improvement

NextToken helps AI product teams continuously improve models after deployment by creating a feedback loop from production to the model, turning general-purpose AI into specialized high-performance agents. Its platform converts production traces, expert feedback, and real-world tasks into structured evaluations and training data, then applies post-training methods like supervised fine-tuning, RL with verifiable rewards, policy optimization, and model distillation. It differentiates itself with an end-to-end workflow that ties production feedback directly to model updates, enabling organization-specific intelligence and practical deployment of optimized agents. Goal: help teams deploy and steadily improve specialized AI models with high accuracy in real-world tasks.

Company size
1-10
Company stage
Not listed
Total funding
Not listed
Headquarters
Austin, TX
Founded
2016

Recently Posted Jobs

Explore more opportunities at NextToken

See more jobs