AMI Research Engineer

The role

You will work with a team of scientists and engineers to advance the state-of-the-art in world modeling. You will implement self-supervised learning methods, develop new architectures, and create scalable infrastructure and algorithms.

What you will do

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

Required qualifications

• Bachelor’s degree or equivalent experience in Computer Science or a related field
• Proficiency in Python
• Ability to design, run, and analyze experiments independently
• Understanding of machine learning fundamentals, large-scale training, and accelerator-based (GPU or TPU) compute environments

Nice to have

• Strong track record of building and deploying high-performance ML models
• Experience developing, testing, and maintaining large-scale distributed systems
• Experience releasing and maintaining open-source projects
• Proficiency in a deep learning framework (PyTorch or JAX), especially for distributed training and efficient inference

About the company

ami is a global company with offices in Paris, Montreal, New York, and Singapore. They are a team of scientists and engineers building frontier world model-based AI.