AI Engineer – World Modeling (Singapore)

About the role

AMI is building next‑generation AI systems that understand the real world, retain persistent memory, reason, plan, and remain safe. As an AI Engineer in Singapore you will join a global research‑focused team to advance world‑model research using video and multimodal data.

Key responsibilities

• Develop self‑supervised learning methods for video and high‑dimensional sensor data.
• Design new architectures that predict world dynamics from continuous signals.
• Build scalable pipelines for preprocessing and curating large video datasets.
• Create evaluation frameworks to benchmark model understanding, prediction, and planning.
• Implement efficient algorithms for model‑based planning and reasoning.

Required profile

• Bachelor’s degree or equivalent experience in Computer Science or a related field.
• Proficiency in Python and ability to design, run, and analyse experiments independently.
• Solid understanding of machine‑learning fundamentals, large‑scale training, and GPU/TPU compute environments.

Required skills

• Python
• PyTorch or JAX
• GPU and TPU acceleration

What we offer

• Opportunity to work on cutting‑edge AI research with a world‑class team.
• Collaborative environment across offices in Paris, Montreal, New York, and Singapore.
• Focus on engineering excellence and execution of frontier AI technologies.