bundle_adjust: Add option to specify loss function