Google TPU optimizations for transformers models - View it on GitHub