A transient error might happen at any point on a multi statement transaction, and the client should be the responsible of retrying in the presence of such errors (like for example a StaleConfig). Something similar to SERVER-58145 could be done, using the auto_retry_transaction_in_sharding.js helper.