Member of Technical Staff, Reinforcement Learning

Inception Labs
San Mateo
Workplace: OnsiteFull timeFunction: Education & TrainingExperience: 2+ yearsSkills: ["Reinforcement learning","PyTorch","Transformers","Diffusion models","RLHF","PPO","DPO","TensorRT","LLM serving"]

Role focusing on designing and implementing reinforcement learning pipelines for diffusion-based LLMs, developing reward models, and aligning model behavior with human intent at scale. You will optimize RL training (PPO, DPO, RLHF), work on data preprocessing and evaluation, and advance techniques for controlled text generation and scalable AI systems in a cutting-edge research environment.

Loading

Loading job details...

Preparing the role view and application actions.

FursaFursa
Inception Labs
Inception Labs
6 months ago

Member of Technical Staff, Reinforcement Learning

✓ Verified Job

Canonical indexed version, validated from employer's careers page.

Source: Company careers pageValidated by: Fursa AI
Last checked: 3 hours agoStatus: Live

Job Summary

Role focusing on designing and implementing reinforcement learning pipelines for diffusion-based LLMs, developing reward models, and aligning model behavior with human intent at scale. You will optimize RL training (PPO, DPO, RLHF), work on data preprocessing and evaluation, and advance techniques for controlled text generation and scalable AI systems in a cutting-edge research environment.
Location: San Mateo
Workplace: Onsite
Employment Type: Full time
Job Function: Education & Training

Key Responsibilities

  • •Design, develop, and optimize RL training pipelines (PPO, DPO, RLHF, and novel approaches) for diffusion-based LLMs
  • •Build and iterate on reward models, reward shaping strategies, and evaluation of reward quality
  • •Implement innovative approaches for fine-tuning and scaling generative AI models
  • •Work on data preprocessing pipelines, model evaluation, and alignment to enterprise use cases
  • •Research and implement techniques for controlled text generation and constraint satisfaction
  • •Improve training stability, efficiency, and reproducibility of RL workloads

Pay and Benefits

Equity and Bonus:Equity
Perks:Health InsuranceDentalVisionEquityMeal AllowanceCommuter BenefitsPaid Leave

Key Requirements

  • •BS/MS/PhD in Computer Science or a related field (or equivalent experience)
  • •At least 2 years of experience working on ML projects in PyTorch (or equivalent)
  • •Excellent familiarity with transformers and core LLM concepts (autoregressive pretraining, instruction tuning, in-context learning, KV caching)
  • •Hands-on experience with reinforcement learning from human feedback (RLHF), PPO, DPO, or related post-training methods
  • •Familiarity with training and inference in diffusion models
Experience:2+ yearsAIMLLLMNLP
Skills:Reinforcement learningPyTorchTransformersDiffusion modelsRLHFPPODPOTensorRTLLM serving
Tech Stack:PyTorchTransformersDiffusion modelsRLHFPPODPOTensorRT

Company Brief

Inception Labs
Develops artificial intelligence solutions and research-driven products, focusing on machine learning models, AI tools, and enterprise AI integrations to help organizations automate workflows, extract insights from data, and build intelligent applications.
Industry: AI & Machine Learning
Website