Member of Technical Staff - Machine Learning, Frontier AI Robotics

Amazon Amazon · Big Tech · San Francisco, CA · Software Development

Leads an ML infrastructure team focused on creating model training and simulation environments for large robotics foundation models. This involves defining roadmaps, building realistic simulation environments for RL and synthetic data generation, and implementing tooling for data creation and experimentation. The role emphasizes large-scale training, multi-modal models, and robotics applications.

What you'd actually do

  1. 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.
  2. Work together with AI researchers to implement and optimize training of new model architectures at a large scale
  3. Define roadmap and build physics realistic simulation environment for reinforcement learning, closed-loop simulations and synthetic data generation.
  4. Implement tooling for data creation, model experimentation, and continuous integration

Skills

Required

  • Python
  • C++
  • PyTorch
  • building large-scale training infrastructure
  • Isaac Sim, Unity, Unreal or proprietary 3D game engine, or industry-equivalent technology (3D animation, simulation, etc)
  • planning, designing, developing and delivering infrastructure software

Nice to have

  • designing and developing large scale, high-traffic applications
  • physical robots
  • reinforcement learning
  • synthetic data generation
  • optimizing physics simulation for articulated robots and rigid body interactions
  • VLM
  • Imitation learning
  • VLA
  • implementing techniques from research papers

What the JD emphasized

  • 10+ years of engineering experience
  • 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
  • Experience with physical robots, reinforcement learning, synthetic data generation.

Other signals

  • large-scale training infrastructure
  • multi-modal robotic foundation models
  • reinforcement learning
  • synthetic data generation