Member of Technical Staff — Post-Training Infrastructure
Embedding VC20 days ago
Responsibilities
- Build reinforcement learning 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’ planning, coding, tool use, error recovery, and complex workflow completion.
- Scale distributed training and high-throughput rollout generation across multi-GPU environments.
- Improve training reliability, reproducibility, observability, and cost efficiency.
- Help define long-term strategies for agent, robotics, and embodied-model training.
Requirements
- 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 reinforcement learning.
- 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 includes work at a frontier AI lab, code-model post-training, autonomous coding agents, multimodal models, robotics, simulation, embodied AI, verifiable rewards, outcome-based training, or large GPU clusters.
Benefits
- On-site, in-person work in San Francisco.
- Opportunity to build foundational post-training and reinforcement learning infrastructure for agents, robotics, simulation, and embodied AI.
Categories
About Embedding VC
Embedding VC is an early-stage venture fund that backs generative AI startups, led by current and former builders and operators. The partnership invests capital and hands-on expertise to help AI founders scale, with a focus on progress toward AGI. Founded in 2023 and headquartered in Menlo Park, California, it operates as a private partnership concentrating on AI-native applications and infrastructure.