adaptive computation time

Todo

https://arxiv.org/pdf/1603.08983