Universal Transformer with dynamic halting