Hedra is a pioneering generative modeling company looking for a Research Engineer to lead pre-training and post-training on action-conditioned world models and work with industrial partners to adapt generative models for real-world physical AI applications.
Requirements
- Design, implement, and run pre-training and post-training pipelines for action-conditioned world models and vision-language-action (VLA) models
- Develop and refine training methodologies, including fine-tuning, reinforcement learning, and large-scale multimodal learning
- Design and generate training and evaluation datasets from simulation, including environment setup, domain randomization, and sim-to-real transfer strategies
- Build distributed training infrastructure using PyTorch, FSDP, and DeepSpeed
- Work with multimodal data pipelines involving video, sensory inputs, and action sequences
- Evaluate model performance using both benchmark datasets and real-world deployment metrics
- Collaborate with industrial partners to adapt generative models for real-world physical AI applications
Benefits
- Competitive compensation
- Equity
- 401k
- Healthcare (Silver PPO Medical, Vision, Dental)
- Lunch and snacks at the office