AI Engineer – World Modeling

About the role

AMI is building AI agents that understand the real world through world models. As an AI Engineer you will help advance state‑of‑the‑art world modeling using video and other high‑dimensional sensor data.

Key responsibilities

• Implement and optimise robust, scalable self‑supervised learning methods for video and continuous signals.
• Design and scale new architectures that predict world dynamics efficiently.
• Build scalable infrastructure and algorithms for video data pre‑processing and curation.
• Develop efficient model‑based planning and reasoning algorithms for complex tasks.

Required profile

• Bachelor’s degree in Computer Science or a related field, or equivalent experience.
• 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
• Deep‑learning frameworks: PyTorch or JAX
• Experience with distributed training and large‑scale systems
• GPU/TPU acceleration
• Self‑supervised learning techniques
• Video data processing and world‑model prediction