World-Model AI Engineer — Video & Planning

ami; Singapore; On-site; SGD 90,000 - 140,000; Full time

Job summary

AMI is building a new breed of AI systems with persistent memory and controllable, safe behavior. You will join a world-model research team in Singapore, focusing on self-supervised learning from video and high-dimensional signals, as well as architectures that predict world dynamics and enable planning.

We seek researchers who can design, run and analyze experiments, and who have strong Python and ML fundamentals. Experience with PyTorch or JAX and open-source contributions is a plus.

Qualifications

• Bachelor’s degree in CS or a related field.
• Proficiency in Python and ML fundamentals.
• Ability to design, run and analyze experiments independently.

Responsibilities

• Collaborate on world model research efforts.
• Advance self-supervised learning methods from video and high-dimensional signals.
• Develop architectures that predict world dynamics from video and other signals.
• Create scalable data pre-processing and curation pipelines for video data.
• Contribute to evaluations benchmarking world model understanding, prediction, and planning.
• Work on efficient algorithms for model-based planning and reasoning.

Skills

Python

Experiment design

ML fundamentals

Independent work

Multimodal models

Planning algorithms

Education

Bachelor’s degree in Computer Science or related field

Tools

PyTorch

JAX

Job description

AMI is building a new breed of AI systems with persistent memory and controllable, safe behavior. You will join a world-model research team in Singapore, focusing on self-supervised learning from video and high-dimensional signals, as well as architectures that predict world dynamics and enable planning.

We seek researchers who can design, run and analyze experiments, and who have strong Python and ML fundamentals. Experience with PyTorch or JAX and open-source contributions is a plus.