A

Amazon

Sells products online and rents cloud computing

Applied Scientist 2 - Robotics Simulation

Applied Machine LearningConfirmed live in the last 24 hours
Full-timeIn-person

About the Job

Requirements

  • Hold a PhD or Master's degree.
  • Have knowledge of machine learning frameworks including JAX, PyTorch, vLLM, SGLang, Dynamo, TorchXLA, and TensorRT.
  • Have experience in robotics design, automation systems development, control systems design, or related product development.
  • Have at least 2 years of experience working with physics simulation platforms for robot learning, such as MuJoCo, Isaac Sim/Lab, PyBullet, or Drake.
  • Demonstrate experience training robot policies using reinforcement learning or imitation learning and evaluating them in simulation.
  • Have experience with articulated robot simulation, including URDF/MJCF/USD formats and rigid/soft body dynamics.
  • Be familiar with sim-to-real transfer concepts, including domain randomization, system identification, or physics calibration.

Responsibilities

  • Design and implement GPU-accelerated reinforcement learning and imitation learning environments in NVIDIA Isaac Lab for manipulation and mobility tasks.
  • Build and maintain policy training pipelines supporting diffusion policies, vision-language-action models, behavior cloning, and actor-critic reinforcement learning, and evaluate trained policies in simulation.
  • Characterize and reduce sim-to-real gaps by comparing simulated sensor outputs, kinematics, and dynamics against real-world robot data, then implementing targeted improvements.
  • Implement visual, physics, and geometric domain randomization strategies to improve policy robustness and transfer to real hardware.
  • Develop sim-to-real transfer techniques, including system identification, physics parameter calibration, and visual domain adaptation.
  • Create robot embodiment validation tests for joint kinematics, actuator response, and contact behavior to ensure digital twins are faithful to real hardware.
  • Build data pipelines for recording, replaying, and augmenting demonstration data from teleoperation or automated trajectory generation to scale training data volume.
  • Contribute to end-effector modeling and contact dynamics tuning to ensure physically plausible gripper and tool interactions in simulation.
  • Author design documents for new simulation science capabilities and contribute to technical reviews.
  • Collaborate with partner science teams to understand their model architectures and ensure simulation environments meet their training requirements.
  • Own simulation science deliverables end-to-end, from environment design through policy evaluation, with increasing scope and independence.
  • Contribute to technical design discussions, propose improvements to simulation fidelity and training methodology, and help establish best practices for robot learning in simulation.

Desired Qualifications

  • Have hands-on experience deploying learned policies on real robot hardware, such as manipulation arms, mobile platforms, or mobile manipulators.
  • Have experience with NVIDIA Isaac Lab/Sim, Omniverse, or USD-based simulation workflows.
  • Have experience with modern Physical AI architectures, including vision-language-action models, diffusion-based policy learning, action-chunking transformers, or behavior cloning from demonstrations.
  • Be familiar with teleoperation systems and demonstration data collection pipelines, including haptic devices and recording in HDF5/zarr or similar formats.
  • Have experience with contact dynamics modeling and parameter tuning for grippers, suction systems, or other end-effectors.
  • Be familiar with ROS2 and robotics middleware for integration with robot software stacks.
  • Have experience with GPU-accelerated parallel simulation, including running thousands of environments concurrently for reinforcement learning training.
  • Have experience with trajectory optimization or motion planning, such as MoveIt or OMPL, for generating training data.
  • Publications at robotics or machine learning venues such as CoRL, ICRA, IROS, RSS, NeurIPS, or ICML are a plus but not required.

About the Company

Amazon

Sells products online and rents cloud computing

Amazon runs the largest online store in the United States and rents cloud computing to businesses and governments through Amazon Web Services. AWS pioneered renting computing power over the internet, and Amazon built its own delivery network to get orders to customers quickly. Shoppers buy from Amazon and from independent sellers on its store, and Prime members pay a subscription for fast free shipping, streaming video and music, and other perks. AWS rents out computing power, storage, databases, and AI tools so companies can run software without owning servers, and Amazon also makes Kindle e-readers and Echo speakers with the Alexa assistant. Jeff Bezos founded Amazon in 1994, and it is based in Seattle, Washington.

Company size
10,001+
Company stage
IPO
Headquarters
Seattle, WA
Founded
1994

Recently Posted Jobs

Explore more opportunities at Amazon

See more jobs