TF Train step docstring
# What does this PR do?

Small PR that elaborates on the beautiful black magic behind our TF `train_step`.
_The documentation is not available anymore as the PR was closed or merged._<|||||>branched from the wrong place 🤦 