Research Engineer

Hedra is a pioneering generative modeling company that's building a Physical AI team. As a Research Engineer, you'll lead pre-training and post-training on action-conditioned world models, working with industrial partners to bring generative AI to real-world 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
• Build distributed training infrastructure using PyTorch, FSDP, and DeepSpeed

Benefits

• Competitive compensation
• Equity
• 401k
• Healthcare
• Lunch and snacks at the office