AMI Research Engineer

What you'll do

• Implement and optimize scalable self-supervised learning methods for video and other continuous signals.
• Develop efficient architectures that predict world dynamics from video and other high-dimensional signals.
• Build scalable video data preprocessing and curation infrastructure, and algorithms for model-based planning and reasoning.

What they ask for

• 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 GPU or TPU compute environments.

Nice to have

• Track record of building and deploying high-performance ML models; experience with large-scale distributed systems or open-source projects.
• Proficiency in PyTorch or JAX, especially for distributed training and efficient inference.

Tools

Python, GPU, TPU, PyTorch, JAX

AI focus

Research and engineering for self-supervised world models that learn from video and other high-dimensional signals, predict world dynamics, and enable model-based planning and reasoning.