Implementation of a Transformer that Ponders, using the scheme from the PonderNet paper - View it on GitHub