Member of Technical Staff AI Researcher

You will research, train, and optimize large language models using supervised and reinforcement learning. You will build data, training, and evaluation pipelines, develop training frameworks, run experiments, integrate models into products, and translate user needs into model improvements.

Responsibilities

• Post-train LLMs using supervised and reinforcement learning techniques
• Scale model performance using query and answer data
• Research model training, optimization, and personalization techniques
• Implement preference optimization and personalization capabilities
• Invent model improvements and optimizations
• Turn research ideas into algorithms, run experiments, and launch models
• Own data, training, and evaluation pipelines
• Build LLM post-training frameworks on Megatron and PyTorch
• Implement infrastructure for large-scale model training
• Integrate models into products
• Partner with engineering and product teams on model improvements

Requirements

• Large-scale LLM and deep learning systems
• Python
• PyTorch
• Post-training techniques
• Reinforcement learning
• 2–6 years of relevant project experience

Benefits

• Equity
• Health insurance
• Dental insurance
• Vision insurance
• Retirement benefits
• Fitness account
• Commuter account
• Dependent care account