dvsaisurya

### Workload LM1B #### Task Transformer language model training, resembling https://github.com/google/flax/tree/main/examples/lm1b . #### Dataset LM1B dataset. https://www.tensorflow.org/datasets/catalog/lm1b #### Model decoder-only Transformer, encoder+decoder based Transformer. #### Reference Implementation https://github.com/google/flax/tree/main/examples/lm1b --- ###...