Member of Technical Staff — Post-Training Infrastructure

San Francisco Bay Areafull time

Apply on Employer Site

On Site

senior

H1B: no

Embedding VC Member of Technical Staff at Embedding VC to lead reinforcement learning infrastructure and model post-training for AI agents. Role involves building RL pipelines and scaling distributed training.

Responsibilities

We are hiring a Member of Technical Staff to lead reinforcement learning infrastructure and model post-training.
Moonlake already has deep expertise in 3D and world-building. This role adds the model-training experience needed to systematically improve agent performance and prepare the company for larger-scale RL across both digital and physical environments.
We are looking for a full-stack researcher and engineer who understands the complete training system and can make strong judgments about when training is necessary, which methods are likely to work, and what not to pursue.
Build RL and post-training pipelines for multimodal, vision-language, and code-generating agents
Develop infrastructure for supervised fine-tuning, preference optimization, reward modeling, and reinforcement learning
Create systems for collecting, filtering, replaying, and learning from agent trajectories
Design rewards, verifiers, and evaluations for long-horizon agent tasks
Improve agents’ ability to plan, write and execute code, use tools, recover from errors, and complete complex workflows
Scale distributed training and high-throughput rollout generation across multi-GPU environments
Improve training reliability, reproducibility, observability, and cost efficiency
Help define Moonlake’s long-term strategy for agent, robotics, and embodied-model training

Qualification

Familiarity with supervised fine-tuningExperience with code-generation agentsAbility to work across dataExperience with multimodal modelsWhat Success Looks LikeWithin your first year

Required

Real-world experience training large language, vision-language, multimodal, or code models
Strong experience in reinforcement learning, post-training, or large-scale fine-tuning
Experience building distributed training or high-throughput inference systems
Familiarity with supervised fine-tuning, preference optimization, reward modeling, and agentic RL
Experience with code-generation agents, long-horizon evaluation, or tool-using systems
Strong Python skills and experience with PyTorch, JAX, or similar frameworks
Ability to work across data, models, environments, rewards, evaluation, and infrastructure
Strong research judgment and a bias toward building reliable systems

Preferred

Experience at a frontier AI lab or organization operating large-scale training systems
Experience with code-model post-training or autonomous coding agents
Experience with multimodal models, robotics, simulation, or embodied AI
Experience designing verifiable rewards or outcome-based training systems
Experience scaling RL workloads across large GPU clusters
What Success Looks Like
Within your first year, you will have:
Built Moonlake’s core post-training and RL infrastructure

Benefits

Vision

Company

Embedding VC

View company page

Source: ashby

Active within the past hour

More roles at Embedding VC

Member of Technical Staff, Infra

Mountain View, CA · Remote

Senior/Staff Software Engineer, Agent

San Francisco Bay Area

Senior Software Engineer, Enterprise AI Platform

San Francisco Bay Area

Marketing Data Scientist

San Francisco Bay Area

Boost Your Interview Chances

Improve Resume Match Score

Free

Your Score

9.0

Strong Applicant

Must-Have Skills for This Role

pythonpytorch

Optimize my Resume

Apply Faster with Autofill

Free

Save your profile once. LumiJobs pre-fills the employer's application form so you stop re-typing the same answers on every site.

Apply With Autofill

Or skip LumiJobs's tools and submit directly on the employer's career site.

Apply on Employer Site

Member of Technical Staff — Post-Training Infrastructure at Embedding VC · LumiJobs