Member of Technical Staff - World Models

Veeda AI; Zürich; On-site; CHF 120,000 - 180,000; Full time

Job summary

Veeda AI in Zürich, Switzerland is seeking a Member of Technical Staff to advance world models for physical AI. You will design, train, and scale multimodal foundation models, implement distributed training in PyTorch, and collaborate with robotics engineers to translate research into reliable capabilities.

The ideal candidate holds a PhD in CS or equivalent, has proven end-to-end training experience, and can design rigorous experiments to inform research decisions.

Qualifications

• PhD degree in Computer Science, Engineering, or a related technical field, or equivalent experience.
• Experience training machine learning foundation models end to end, with strong PyTorch skills and hands-on experience in distributed training.
• Ability to independently design, execute, and analyze machine learning experiments, from hypothesis through ablation to conclusion.
• Experience developing evaluations that reveal model limitations and inform research decisions.

Responsibilities

• Design, train, and scale generative and predictive models across visual and spatial modalities.
• Develop methods for learning from diverse data sources and incorporating conditioning signals into model behavior.
• Improve consistency, robustness, and generalization in models of complex temporal and spatial structure.
• Explore post-training and model optimization techniques to improve quality, controllability, and computational efficiency.
• Develop rigorous evaluations, conduct controlled experiments, and use findings to guide modeling and data decisions.
• Work closely with data, systems, and robotics engineers to translate research ideas into reliable capabilities.

Skills

PyTorch

Distributed training

Multimodal models

Research

Education

PhD in Computer Science or related field

Tools

Python

Job description

Member of Technical Staff – World Models

About Us

Veeda AI is building the next generation of multimodal foundation world models for Physical AI. We're a small, fast-moving team of engineers and researchers from leading AI labs, tackling some of the most challenging problems at the intersection of AI, robotics, and embodied intelligence. If you're excited about pushing the boundaries of what's possible with Physical AI, you'll have the opportunity to make an outsized impact from day one.

Responsibilities

• Model Research & Development: Design, train, and scale generative and predictive models across visual and spatial modalities.
• Multimodal Learning & Control: Develop methods for learning from diverse data sources and incorporating conditioning signals into model behavior.
• Sequence Modeling: Improve consistency, robustness, and generalization in models of complex temporal and spatial structure.
• Post-Training & Efficiency: Explore post-training and model optimization techniques to improve quality, controllability, and computational efficiency.
• Evaluation & Experimentation: Develop rigorous evaluations, conduct controlled experiments, and use findings to guide modeling and data decisions.
• Research Collaboration: Work closely with data, systems, and robotics engineers to translate research ideas into reliable capabilities.

Requirements

• PhD degree in Computer Science, Engineering, or a related technical field, or equivalent experience.
• Experience training machine learning foundation models end to end, with strong PyTorch skills and hands-on experience in distributed training.
• Ability to independently design, execute, and analyze machine learning experiments, from hypothesis through ablation to conclusion.
• Experience developing evaluations that reveal model limitations and inform research decisions.

Nice to Have

• Strong publication record or substantial contributions to research on image, video, 3D, or multimodal models.
• Experience with learned representations, compression, or tokenization for high-dimensional data.
• Experience with conditional generation, temporal consistency, or long-context modeling.
• Experience with reinforcement learning, model-based learning, or embodied AI.
• Experience with post-training, distillation, or efficient inference for generative models.
• Experience working with large-scale, heterogeneous visual or sensor datasets.
• Contributions to open-source machine learning projects or research infrastructure.