Research Engineer

Hedra is a pioneering generative modeling company building a Physical AI team to apply generative models to real-world industry and economy use cases. As a Research Engineer, you will lead pre-training and post-training on action-conditioned world models, collaborating with industrial partners to connect generative AI with physical systems.

Responsibilities:

• 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
• Contributions research publications a plus
• Collaborate with industrial partners to adapt generative models for real-world physical AI applications