Amazon is looking to develop robust physical AI solutions for dynamic real-world environments by building large robotics foundation models and efficient, promptable model architectures.
Requirements
- Proficiency in Python and C++, experience in building large-scale training infrastructure using PyTorch
- Experience with Isaac Sim, Unity, Unreal or proprietary 3D game engine, or industry-equivalent technology (3D animation, simulation, etc)
- Experience with physical robots, reinforcement learning, synthetic data generation.
- Experience optimizing physics simulation for articulated robots and rigid body interactions.
- Deep understanding of different model architecture like VLM, Imitation learning, VLA
- Experience implementing techniques from research papers
Responsibilities
- You will lead the ML infrastructure team to create model training and simulation environment for developing large robotics foundational models for reasoning, perception, locomotion, and manipulation.
- Work together with AI researchers to implement and optimize training of new model architectures at a large scale
- Define roadmap and build physics realistic simulation environment for reinforcement learning, closed-loop simulations and synthetic data generation.
- Implement tooling for data creation, model experimentation, and continuous integration
Other
- 5+ years of engineering team management experience
- Experience managing multiple concurrent programs, projects and development teams in an Agile environment
- Experience partnering with product or program management science teams working in research environments
- work safely and cooperatively with other employees, supervisors, and staff
- adhere to standards of excellence despite stressful conditions
