Amazon's Frontier AI & Robotics team is looking to develop breakthrough foundation models that enable robots to perceive, understand, and interact with the world in unprecedented ways, pushing the boundaries of robotic intelligence and enabling real-world deployment at Amazon scale.
Requirements
- Experience programming in Java, C++, Python or related language
- Experience building machine learning models or developing algorithms for business application
- Experience applying theoretical models in an applied environment
- Experience in state-of-the-art deep learning models architecture design and deep learning training and optimization and model pruning
- Experience using Unix/Linux
- Experience with foundation models or large language models
- Extensive programming skills in Python and PyTorch/JAX
Responsibilities
- Design and implement novel deep learning architectures that push the boundaries of what robots can understand and accomplish
- Drive independent research initiatives in robotics foundation models, focusing on breakthrough approaches in perception, and manipulation, for example open-vocabulary panoptic scene understanding, scaling up multi-modal LLMs, sim2real/real2sim techniques, end-to-end vision-language-action models, efficient model inference, video tokenization
- Lead technical projects from conceptualization through deployment, ensuring robust performance in production environments
- Collaborate with platform teams to optimize and scale models for real-world applications
- Contribute to the team's technical strategy and help shape our approach to next-generation robotics challenges
- Design and implement novel foundation model architectures, leveraging our extensive compute infrastructure to train and evaluate at scale
- Collaborate with our world-class research team to solve complex technical challenges
Other
- PhD, or Master's degree and 4+ years of CS, CE, ML or related field experience
- Experience in patents or publications at top-tier peer-reviewed conferences or journals
- Track record of solving complex technical problems
- First-author publications at major or top-tier ML / robotics conferences (NeurIPS, ICML, ICLR, CVPR, ICCV, RSS, CoRL)
- Background in computer vision, robotics, or related fields
