Legged gym paper. You can use any reward function defined in legged_robot.