Abstract
One embodiment of a method for controlling a robot includes generating, via a first trained machine learning model, a robot motion and a predicted force associated with the robot motion, determining, via a second trained machine learning model, a gain associated with the predicted force, generating one or more robot commands based on the robot motion and the gain, and causing a robot to move based on the one or more robot commands.