This is unofficial implementation of Universal Transformer arXiv:1807.03819 by TensorFlow Eager Execution and Keras API. The official implementation is available in tensor2tensor.
- beam search
- switch UT and Adaptive UT by config
- example notebook
- example notebook for algorithmic task
Apache License 2.0