World-Model AI Engineer: Video Forecasting & Planning

AMI Labs; Paris; Sur place; EUR 80 000 - 120 000; Plein temps

Résumé du poste

AMI Labs in Paris, France, is hiring for a role focusing on advancing world model-based AI by predicting and planning with internal world models. You will work on self-supervised learning from video and high-dimensional signals to understand context and forecast dynamics, enabling robust planning in dynamic environments.

You will collaborate with scientists and engineers to develop scalable architectures, pre-processing pipelines for video data, and efficient inference on GPU/TPU clusters,

Qualifications

• Bachelor’s degree or equivalent in Computer Science or related field
• Proficiency in Python
• Ability to design, run, and analyze experiments independently
• Understanding of machine learning fundamentals and large-scale training
• Experience with PyTorch or JAX and distributed training

Responsabilités

• Implement robust and scalable self-supervised learning methods from video and high-dimensional signals
• Develop and scale architectures to predict world dynamics from video
• Create scalable infrastructure for pre-processing and curating video data
• Design efficient algorithms for model-based planning and reasoning

Connaissances

Python proficiency

Experiment design & analysis

Machine learning fundamentals

Distributed training

Formation

Bachelor’s degree in Computer Science or related field

Outils

PyTorch

JAX

CUDA / GPU compute

Description du poste

AMI Labs in Paris, France, is hiring for a role focusing on advancing world model-based AI by predicting and planning with internal world models. You will work on self-supervised learning from video and high-dimensional signals to understand context and forecast dynamics, enabling robust planning in dynamic environments.

You will collaborate with scientists and engineers to develop scalable architectures, pre-processing pipelines for video data, and efficient inference on GPU/TPU clusters,