Research Engineer: Action-Conditioned World Models

Hedra, Inc; San Francisco (CA); On-site; USD 120,000 - 160,000; Full time

Benefits offered by this job

Competitive compensation and equity

401k (no match)

Healthcare (Silver PPO Medical, Vision, Dental)

Lunch and snacks at the office

Job summary

A pioneering generative modeling company in San Francisco is seeking a Research Engineer for their Physical AI team. This role focuses on designing pre-training and post-training pipelines for AI models, collaborating with industrial partners, and evaluating model performance. Candidates should have experience in large generative models, proficiency in PyTorch, and a relevant degree. The position offers competitive compensation, healthcare, and snacks at the office.

Qualifications

• Experience with pre-training or post-training on large generative models.
• Hands-on proficiency with PyTorch and distributed training frameworks.
• Strong fundamentals in machine learning, optimization, and large-scale data processing.

Responsibilities

• Design, implement, and run pre-training and post-training pipelines.
• Collaborate with partners to adapt generative models for real-world applications.
• Evaluate model performance using benchmark datasets and real-world metrics.

Skills

Pre-training on generative models

Distributed training frameworks (FSDP, DeepSpeed)

Machine learning fundamentals

Optimization techniques

Multimodal data processing

Video understanding

Education

BS/MS/PhD in Computer Science, Machine Learning, Robotics

Tools

PyTorch

Job description

A pioneering generative modeling company in San Francisco is seeking a Research Engineer for their Physical AI team. This role focuses on designing pre-training and post-training pipelines for AI models, collaborating with industrial partners, and evaluating model performance. Candidates should have experience in large generative models, proficiency in PyTorch, and a relevant degree. The position offers competitive compensation, healthcare, and snacks at the office.